- Sort Score
- Result 10 results
- Languages All
Results 1 - 4 of 4 for TFE_TensorHandleDim (0.24 sec)
-
tensorflow/c/eager/custom_device_testutil.cc
explicit LoggedTensor(TFE_TensorHandle* tensor) : tensor(tensor) {} ~LoggedTensor() { TFE_DeleteTensorHandle(tensor); } }; int64_t LoggedTensorDim(void* data, int dim_index, TF_Status* status) { return TFE_TensorHandleDim(reinterpret_cast<LoggedTensor*>(data)->tensor, dim_index, status); } int LoggedTensorNumDims(void* data, TF_Status* status) {
C++ - Registered: Tue Apr 23 12:39:09 GMT 2024 - Last Modified: Wed Mar 03 20:47:31 GMT 2021 - 8.3K bytes - Viewed (0) -
tensorflow/c/eager/c_api.h
TF_Status* status); // This function will block till the operation that produces `h` has completed. TF_CAPI_EXPORT extern int64_t TFE_TensorHandleDim(TFE_TensorHandle* h, int dim_index, TF_Status* status);
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/c_api.cc
return -1; } int64_t num_elements = -1; status->status = tensorflow::unwrap(h)->NumElements(&num_elements); return num_elements; } int64_t TFE_TensorHandleDim(TFE_TensorHandle* h, int dim_index, TF_Status* status) { if (h == nullptr) { status->status = tensorflow::errors::InvalidArgument("Invalid handle"); return -1; }
C++ - Registered: Tue Apr 23 12:39:09 GMT 2024 - Last Modified: Tue Mar 12 20:00:09 GMT 2024 - 43.9K bytes - Viewed (2) -
tensorflow/c/eager/c_api_test.cc
ASSERT_EQ(TF_INVALID_ARGUMENT, TF_GetCode(status.get())); ASSERT_EQ(num_dims, -1); ASSERT_EQ("Invalid handle", string(TF_Message(status.get()))); TF_SetStatus(status.get(), TF_OK, ""); int dim = TFE_TensorHandleDim(h, 0, status.get()); ASSERT_EQ(TF_INVALID_ARGUMENT, TF_GetCode(status.get())); ASSERT_EQ(dim, -1); ASSERT_EQ("Invalid handle", string(TF_Message(status.get()))); } TEST(CAPI, TensorHandleDevices) {
C++ - Registered: Tue Apr 23 12:39:09 GMT 2024 - Last Modified: Thu Aug 03 20:50:20 GMT 2023 - 94.6K bytes - Viewed (1)