Searched refs:TF_Dim (Results 1 – 13 of 13) sorted by relevance
| /external/tensorflow/tensorflow/c/kernels/ |
| D | bitcast_op.cc | 102 TF_Dim(tensor, dim_count - 1) == k->out_size / k->in_size))) { in BitcastOp_Compute()
114 dims[dim] = TF_Dim(tensor, dim); in BitcastOp_Compute()
|
| D | summary_op.cc | 118 if (TF_Dim(tensor1, d) != TF_Dim(tensor2, d)) { in IsSameSize()
|
| D | tensor_shape_utils.cc | 32 int64_t dim = TF_Dim(tensor, i); in ShapeDebugString()
|
| /external/tensorflow/tensorflow/c/ |
| D | tf_tensor.h | 121 TF_CAPI_EXPORT extern int64_t TF_Dim(const TF_Tensor* tensor, int dim_index);
|
| D | tf_tensor.cc | 153 int64_t TF_Dim(const TF_Tensor* t, int dim_index) { in TF_Dim() function
165 result *= TF_Dim(t, dim); in TF_TensorElementCount()
|
| D | c_api_test.cc | 141 EXPECT_EQ(dims[0], TF_Dim(t, 0)); in TEST()
142 EXPECT_EQ(dims[1], TF_Dim(t, 1)); in TEST()
166 EXPECT_EQ(dims[0], TF_Dim(t, 0)); in TEST()
167 EXPECT_EQ(dims[1], TF_Dim(t, 1)); in TEST()
1450 EXPECT_EQ(4, TF_Dim(out, 0)); in TEST()
1451 EXPECT_EQ(1, TF_Dim(out, 1)); in TEST()
2314 EXPECT_EQ(dims[i], TF_Dim(value, i)) << i; in TEST_F()
2401 EXPECT_EQ(tensor_dims[i][j], TF_Dim(v, j)) in TEST_F()
|
| D | kernels_test.cc | 988 EXPECT_EQ(dims[i], TF_Dim(tensor, i)); in validate_tensor()
|
| /external/tensorflow/tensorflow/cc/experimental/base/public/ |
| D | tensor.h | 132 return TF_Dim(tensor_.get(), d); in dim_size()
|
| /external/tensorflow/tensorflow/c/experimental/gradients/ |
| D | grad_test_helper.cc | 88 auto dim_analytical = TF_Dim(analytical_tensor, j); in CheckTensorValue()
|
| /external/tensorflow/tensorflow/c/eager/ |
| D | gradient_checker.cc | 42 out_dims[i] = TF_Dim(t, i); in GetDims()
|
| /external/tensorflow/tensorflow/java/src/main/native/ |
| D | tensor_jni.cc | 464 dims[i] = static_cast<jlong>(TF_Dim(t, i)); in Java_org_tensorflow_Tensor_shape()
548 num_elements *= TF_Dim(t, i); in Java_org_tensorflow_Tensor_readNDArray()
|
| /external/tensorflow/tensorflow/go/ |
| D | tensor.go | 202 shape[i] = int64(C.TF_Dim(c, C.int(i)))
|
| /external/tensorflow/tensorflow/python/lib/core/ |
| D | ndarray_tensor.cc | 321 dims->push_back(TF_Dim(tensor, i)); in GetPyArrayDimensionsForTensor()
|