Searched refs:TF_NumDims (Results 1 – 10 of 10) sorted by relevance
| /external/tensorflow/tensorflow/java/src/main/native/ |
| D | tensor_jni.cc | 511 const jsize num_dims = TF_NumDims(t); in Java_org_tensorflow_Tensor_shape()
527 int num_dims = TF_NumDims(t); in Java_org_tensorflow_Tensor_setValue()
545 if (TF_NumDims(t) != 0) { \
566 if (TF_NumDims(t) != 0) { in Java_org_tensorflow_Tensor_scalarBytes()
598 int num_dims = TF_NumDims(t); in Java_org_tensorflow_Tensor_readNDArray()
|
| /external/tensorflow/tensorflow/c/ |
| D | c_api_experimental_test.cc | 49 ASSERT_EQ(0, TF_NumDims(out)); // scalar in TestFakeIteratorStack()
98 ASSERT_EQ(4, TF_NumDims(image)); in TEST()
111 ASSERT_EQ(1, TF_NumDims(label)); in TEST()
|
| D | c_api_test.cc | 137 EXPECT_EQ(2, TF_NumDims(t)); in TEST()
162 EXPECT_EQ(2, TF_NumDims(t)); in TEST()
968 EXPECT_EQ(0, TF_NumDims(out)); // scalar in TEST()
985 EXPECT_EQ(0, TF_NumDims(out)); // scalar in TEST()
1029 EXPECT_EQ(0, TF_NumDims(out)); // scalar in RunMinTest()
1411 EXPECT_EQ(2, TF_NumDims(out)); in TEST()
1851 ASSERT_EQ(0, TF_NumDims(t)); in ScalarFloatFromTensor()
2274 EXPECT_EQ(ndims, TF_NumDims(value)); in TEST_F()
2275 for (int i = 0; i < TF_NumDims(value); ++i) { in TEST_F()
2309 EXPECT_EQ(0, TF_NumDims(t_out)); in TEST_F()
[all …]
|
| D | while_loop_test.cc | 100 EXPECT_EQ(0, TF_NumDims(out)); in ExpectOutputValue()
|
| D | c_api_function_test.cc | 106 EXPECT_EQ(1, TF_NumDims(out)); in RunT()
129 EXPECT_EQ(0, TF_NumDims(out)); // scalar in Run()
1718 EXPECT_EQ(0, TF_NumDims(out)); // scalar in TEST_F()
|
| D | c_api.h | 263 TF_CAPI_EXPORT extern int TF_NumDims(const TF_Tensor*);
|
| D | c_api.cc | 258 int TF_NumDims(const TF_Tensor* t) { return t->shape.dims(); } in TF_NumDims() function
267 int rank = TF_NumDims(t); in TF_TensorElementCount()
|
| /external/tensorflow/tensorflow/c/kernels/ |
| D | bitcast_op.cc | 97 dim_count = TF_NumDims(tensor); in BitcastOp_Compute()
|
| /external/tensorflow/tensorflow/python/lib/core/ |
| D | ndarray_tensor.cc | 274 const int ndims = TF_NumDims(tensor); in GetPyArrayDimensionsForTensor()
|
| /external/tensorflow/tensorflow/go/ |
| D | tensor.go | 143 if ndims := int(C.TF_NumDims(c)); ndims > 0 {
|