- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for TFE_TensorDebugInfoOnDeviceNumDims (0.2 sec)
-
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) {
Registered: Tue Nov 05 12:39:12 UTC 2024 - Last Modified: Mon Apr 06 22:10:09 UTC 2020 - 2.3K bytes - Viewed (0) -
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) {
Registered: Tue Nov 05 12:39:12 UTC 2024 - Last Modified: Sat Oct 12 05:11:17 UTC 2024 - 2.5K bytes - Viewed (0)