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)))
|