Searched refs:TF_Dim (Results 1 – 6 of 6) sorted by relevance
/external/tensorflow/tensorflow/c/ |
D | c_api_test.cc | 135 EXPECT_EQ(dims[0], TF_Dim(t, 0)); in TEST() 136 EXPECT_EQ(dims[1], TF_Dim(t, 1)); in TEST() 160 EXPECT_EQ(dims[0], TF_Dim(t, 0)); in TEST() 161 EXPECT_EQ(dims[1], TF_Dim(t, 1)); in TEST() 1391 EXPECT_EQ(4, TF_Dim(out, 0)); in TEST() 1392 EXPECT_EQ(1, TF_Dim(out, 1)); in TEST() 2076 EXPECT_EQ(dims[i], TF_Dim(value, i)) << i; in TEST_F() 2154 EXPECT_EQ(tensor_dims[i][j], TF_Dim(v, j)) in TEST_F()
|
D | c_api.h | 266 TF_CAPI_EXPORT extern int64_t TF_Dim(const TF_Tensor* tensor, int dim_index);
|
D | c_api.cc | 243 int64_t TF_Dim(const TF_Tensor* t, int dim_index) { in TF_Dim() function
|
/external/tensorflow/tensorflow/java/src/main/native/ |
D | tensor_jni.cc | 508 dims[i] = static_cast<jlong>(TF_Dim(t, i)); in Java_org_tensorflow_Tensor_shape() 604 num_elements *= TF_Dim(t, i); in Java_org_tensorflow_Tensor_readNDArray()
|
/external/tensorflow/tensorflow/python/lib/core/ |
D | ndarray_tensor.cc | 275 dims[i] = TF_Dim(tensor, i); in GetPyArrayDimensionsForTensor()
|
/external/tensorflow/tensorflow/go/ |
D | tensor.go | 151 shape[i] = int64(C.TF_Dim(c, C.int(i)))
|