- Sort Score
- Result 10 results
- Languages All
Results 1 - 4 of 4 for TFE_TensorDebugInfo (0.25 sec)
-
tensorflow/c/eager/c_api_debug.cc
if (!status->status.ok()) { return nullptr; } 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(); }
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) -
tensorflow/c/eager/c_api.h
// // TFE_TensorDebugInfo contains information useful for debugging and // profiling tensors. typedef struct TFE_TensorDebugInfo TFE_TensorDebugInfo; // Retrieves TFE_TensorDebugInfo for `handle`. // If TFE_TensorHandleTensorDebugInfo succeeds, `status` is set to OK and caller // is responsible for deleting returned TFE_TensorDebugInfo.
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) -
tensorflow/c/eager/tfe_tensor_debug_info_internal.h
#ifndef TENSORFLOW_C_EAGER_TFE_TENSOR_DEBUG_INFO_INTERNAL_H_ #define TENSORFLOW_C_EAGER_TFE_TENSOR_DEBUG_INFO_INTERNAL_H_ #include <vector> #include "tensorflow/core/platform/types.h" struct TFE_TensorDebugInfo { explicit TFE_TensorDebugInfo(const std::vector<int64_t>& dims) : dev_dims(dims) {} // Fully-padded, minor-to-major. std::vector<int64_t> dev_dims; };
C - Registered: Tue Apr 23 12:39:09 GMT 2024 - Last Modified: Wed Aug 11 01:20:50 GMT 2021 - 1.1K bytes - Viewed (0) -
tensorflow/c/eager/c_api_debug_test.cc
TFE_Context* ctx = TFE_NewContext(opts, status); CHECK_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status); TFE_DeleteContextOptions(opts); 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));
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)