Searched refs:TFE_TensorHandleNumDims (Results 1 – 9 of 9) sorted by relevance
/external/tensorflow/tensorflow/go/ |
D | tensor_handle.go | 93 n := int(C.TFE_TensorHandleNumDims(th.c, status.c))
|
/external/tensorflow/tensorflow/java/src/main/native/ |
D | eager_operation_jni.cc | 123 int num_dims = TFE_TensorHandleNumDims(tensor_handle, status); in Java_org_tensorflow_EagerOperation_numDims()
|
/external/tensorflow/tensorflow/python/eager/ |
D | pywrap_tensor.cc | 523 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/ |
D | custom_device_testutil.cc | 55 std::vector<int64_t> shape(TFE_TensorHandleNumDims(t->tensor, status)); in MakeLoggedTensorHandle()
|
D | c_api.h | 146 TF_CAPI_EXPORT extern int TFE_TensorHandleNumDims(TFE_TensorHandle* h,
|
D | c_api_distributed_test.cc | 179 EXPECT_EQ(TFE_TensorHandleNumDims(packed_handle, status), 0); in TestFunctionWithPackedInput()
|
D | c_api_test.cc | 535 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()
|
D | c_api.cc | 288 int TFE_TensorHandleNumDims(TFE_TensorHandle* h, TF_Status* status) { in TFE_TensorHandleNumDims() function
|
/external/tensorflow/tensorflow/c/eager/parallel_device/ |
D | parallel_device_test.cc | 207 TFE_TensorHandleNumDims(combined_value.get(), status.get()); in TEST()
|