Searched refs:TF_Dim (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/c/kernels/ |
| D | bitcast_op.cc | 100 TF_Dim(tensor, dim_count - 1) == k->out_size / k->in_size))) { in BitcastOp_Compute()
112 dims[dim] = TF_Dim(tensor, dim); in BitcastOp_Compute()
|
| /external/tensorflow/tensorflow/c/ |
| D | c_api_experimental_test.cc | 99 ASSERT_EQ(batch_size, TF_Dim(image, 0)); in TEST()
100 ASSERT_EQ(224, TF_Dim(image, 1)); in TEST()
101 ASSERT_EQ(224, TF_Dim(image, 2)); in TEST()
102 ASSERT_EQ(3, TF_Dim(image, 3)); in TEST()
112 ASSERT_EQ(batch_size, TF_Dim(label, 0)); in TEST()
|
| D | c_api_test.cc | 138 EXPECT_EQ(dims[0], TF_Dim(t, 0)); in TEST()
139 EXPECT_EQ(dims[1], TF_Dim(t, 1)); in TEST()
163 EXPECT_EQ(dims[0], TF_Dim(t, 0)); in TEST()
164 EXPECT_EQ(dims[1], TF_Dim(t, 1)); in TEST()
1412 EXPECT_EQ(4, TF_Dim(out, 0)); in TEST()
1413 EXPECT_EQ(1, TF_Dim(out, 1)); in TEST()
2276 EXPECT_EQ(dims[i], TF_Dim(value, i)) << i; in TEST_F()
2354 EXPECT_EQ(tensor_dims[i][j], TF_Dim(v, j)) in TEST_F()
|
| D | c_api.h | 267 TF_CAPI_EXPORT extern int64_t TF_Dim(const TF_Tensor* tensor, int dim_index);
|
| D | c_api.cc | 259 int64_t TF_Dim(const TF_Tensor* t, int dim_index) { in TF_Dim() function
269 result *= TF_Dim(t, dim); in TF_TensorElementCount()
|
| /external/tensorflow/tensorflow/java/src/main/native/ |
| D | tensor_jni.cc | 515 dims[i] = static_cast<jlong>(TF_Dim(t, i)); in Java_org_tensorflow_Tensor_shape()
611 num_elements *= TF_Dim(t, i); in Java_org_tensorflow_Tensor_readNDArray()
|
| /external/tensorflow/tensorflow/python/lib/core/ |
| D | ndarray_tensor.cc | 284 dims[i] = TF_Dim(tensor, i); in GetPyArrayDimensionsForTensor()
|
| /external/tensorflow/tensorflow/go/ |
| D | tensor.go | 147 shape[i] = int64(C.TF_Dim(c, C.int(i)))
|