| /external/tensorflow/tensorflow/c/kernels/ |
| bitcast_op.cc | 100 TF_Dim(tensor, dim_count - 1) == k->out_size / k->in_size))) {
112 dims[dim] = TF_Dim(tensor, dim);
|
| /external/tensorflow/tensorflow/c/ |
| c_api_experimental_test.cc | 99 ASSERT_EQ(batch_size, TF_Dim(image, 0));
100 ASSERT_EQ(224, TF_Dim(image, 1));
101 ASSERT_EQ(224, TF_Dim(image, 2));
102 ASSERT_EQ(3, TF_Dim(image, 3));
112 ASSERT_EQ(batch_size, TF_Dim(label, 0));
|
| c_api_test.cc | 138 EXPECT_EQ(dims[0], TF_Dim(t, 0));
139 EXPECT_EQ(dims[1], TF_Dim(t, 1));
163 EXPECT_EQ(dims[0], TF_Dim(t, 0));
164 EXPECT_EQ(dims[1], TF_Dim(t, 1));
[all...] |
| c_api.h | 267 TF_CAPI_EXPORT extern int64_t TF_Dim(const TF_Tensor* tensor, int dim_index);
[all...] |
| c_api.cc | 259 int64_t TF_Dim(const TF_Tensor* t, int dim_index) {
269 result *= TF_Dim(t, dim);
[all...] |
| /external/tensorflow/tensorflow/java/src/main/native/ |
| tensor_jni.cc | 515 dims[i] = static_cast<jlong>(TF_Dim(t, i));
611 num_elements *= TF_Dim(t, i);
|
| /external/tensorflow/tensorflow/go/ |
| tensor.go | 147 shape[i] = int64(C.TF_Dim(c, C.int(i)))
|
| /external/tensorflow/tensorflow/python/lib/core/ |
| ndarray_tensor.cc | 284 dims[i] = TF_Dim(tensor, i);
|
