Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 3 of 3 for TFE_TensorDebugInfoOnDeviceNumDims (0.31 sec)

  1. tensorflow/c/eager/c_api_debug_test.cc

      TFE_TensorHandle* h = TestScalarTensorHandle(ctx, 1.0f);
      TFE_TensorDebugInfo* debug_info = TFE_TensorHandleTensorDebugInfo(h, status);
      CHECK_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status);
    
      ASSERT_EQ(0, TFE_TensorDebugInfoOnDeviceNumDims(debug_info));
    
      TFE_DeleteTensorDebugInfo(debug_info);
      TFE_DeleteTensorHandle(h);
      TFE_DeleteContext(ctx);
      TF_DeleteStatus(status);
    }
    
    TEST(CApiDebug, 2DCPU) {
    C++
    - Registered: Tue Apr 23 12:39:09 GMT 2024
    - Last Modified: Mon Apr 06 22:10:09 GMT 2020
    - 2.3K bytes
    - Viewed (0)
  2. tensorflow/c/eager/c_api_debug.cc

      }
      return new TFE_TensorDebugInfo(dev_dims);
    }
    
    TF_CAPI_EXPORT extern void TFE_DeleteTensorDebugInfo(
        TFE_TensorDebugInfo* debug_info) {
      delete debug_info;
    }
    
    TF_CAPI_EXPORT extern int TFE_TensorDebugInfoOnDeviceNumDims(
        TFE_TensorDebugInfo* debug_info) {
      return debug_info->dev_dims.size();
    }
    
    TF_CAPI_EXPORT extern int64_t TFE_TensorDebugInfoOnDeviceDim(
        TFE_TensorDebugInfo* debug_info, int dim_index) {
    C++
    - Registered: Tue Apr 23 12:39:09 GMT 2024
    - Last Modified: Wed Aug 11 01:20:50 GMT 2021
    - 2.5K bytes
    - Viewed (0)
  3. tensorflow/c/eager/c_api.h

    // different from the number returned by TFE_TensorHandleNumDims.
    // The return value was current at the time of TFE_TensorDebugInfo creation.
    TF_CAPI_EXPORT extern int TFE_TensorDebugInfoOnDeviceNumDims(
        TFE_TensorDebugInfo* debug_info);
    
    // Returns the number of elements in dimension `dim_index`.
    // Tensor representation on device can be transposed from its representation
    C
    - Registered: Tue Apr 23 12:39:09 GMT 2024
    - Last Modified: Thu Apr 27 21:07:00 GMT 2023
    - 22.8K bytes
    - Viewed (1)
Back to top