Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for TFE_TensorDebugInfoOnDeviceDim (0.16 sec)

  1. tensorflow/c/eager/c_api_debug.cc

      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) {
      return debug_info->dev_dims[dim_index];
    }
    
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 05:11:17 UTC 2024
    - 2.5K bytes
    - Viewed (0)
  2. tensorflow/c/eager/c_api.h

    // not the XLA's minor-to-major order.
    // On-device dimensions can be padded. TFE_TensorDebugInfoOnDeviceDim returns
    // the number of elements in a dimension after padding.
    // The return value was current at the time of TFE_TensorDebugInfo creation.
    TF_CAPI_EXPORT extern int64_t TFE_TensorDebugInfoOnDeviceDim(
        TFE_TensorDebugInfo* debug_info, int dim_index);
    
    // Description of the TensorFlow op to execute.
    //
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Thu Apr 27 21:07:00 UTC 2023
    - 22.8K bytes
    - Viewed (0)
Back to top