Home
last modified time | relevance | path

Searched refs:TFE_TensorHandleNumDims (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/go/
Dtensor_handle.go93 n := int(C.TFE_TensorHandleNumDims(th.c, status.c))
/external/tensorflow/tensorflow/java/src/main/native/
Deager_operation_jni.cc123 int num_dims = TFE_TensorHandleNumDims(tensor_handle, status); in Java_org_tensorflow_EagerOperation_numDims()
/external/tensorflow/tensorflow/python/eager/
Dpywrap_tensor.cc523 int n = TFE_TensorHandleNumDims(handle, &self->status); in EagerTensor_shape_tuple()
556 int num_dims = TFE_TensorHandleNumDims(self->handle, &self->status); in EagerTensor_rank()
1075 int num_dims = TFE_TensorHandleNumDims(handle, status.get()); in TFE_Py_TensorShapeSlice()
/external/tensorflow/tensorflow/c/eager/
Dcustom_device_testutil.cc55 std::vector<int64_t> shape(TFE_TensorHandleNumDims(t->tensor, status)); in MakeLoggedTensorHandle()
Dc_api.h146 TF_CAPI_EXPORT extern int TFE_TensorHandleNumDims(TFE_TensorHandle* h,
Dc_api_distributed_test.cc179 EXPECT_EQ(TFE_TensorHandleNumDims(packed_handle, status), 0); in TestFunctionWithPackedInput()
Dc_api_test.cc535 int num_dims = TFE_TensorHandleNumDims(h, status.get()); in TEST()
1225 EXPECT_EQ(0, TFE_TensorHandleNumDims(value_handle, status)); in TEST()
1267 CHECK_EQ(0, TFE_TensorHandleNumDims(h, status)); in BM_ReadVariable()
Dc_api.cc288 int TFE_TensorHandleNumDims(TFE_TensorHandle* h, TF_Status* status) { in TFE_TensorHandleNumDims() function
/external/tensorflow/tensorflow/c/eager/parallel_device/
Dparallel_device_test.cc207 TFE_TensorHandleNumDims(combined_value.get(), status.get()); in TEST()