Searched refs:TF_NumDims (Results 1 – 17 of 17) sorted by relevance
| /external/tensorflow/tensorflow/c/kernels/ |
| D | tensor_shape_utils.cc | 28 CHECK_GE(TF_NumDims(tensor), 0); in ShapeDebugString()
30 for (int i = 0; i < TF_NumDims(tensor); ++i) { in ShapeDebugString()
|
| D | summary_op.cc | 114 if (TF_NumDims(tensor1) != TF_NumDims(tensor2)) { in IsSameSize()
117 for (int d = 0; d < TF_NumDims(tensor1); d++) { in IsSameSize()
|
| D | bitcast_op.cc | 99 dim_count = TF_NumDims(tensor); in BitcastOp_Compute()
|
| D | histogram_summary_op.cc | 82 if (TF_NumDims(safe_tags_ptr.get()) != 0) { in HistogramSummaryOp_Compute()
|
| /external/tensorflow/tensorflow/c/experimental/saved_model/internal/ |
| D | saved_model_api_test.cc | 138 EXPECT_EQ(TF_NumDims(result), 0); in TEST_P()
278 EXPECT_EQ(TF_NumDims(result), 0); in TEST_P()
334 EXPECT_EQ(TF_NumDims(result), 0); in TEST_P()
403 EXPECT_EQ(TF_NumDims(result), 0); in TEST_P()
435 EXPECT_EQ(TF_NumDims(result), 0); in TEST_P()
468 EXPECT_EQ(TF_NumDims(result), 0); in TEST_P()
570 ASSERT_EQ(TF_NumDims(result), 0); in TEST_P()
|
| /external/tensorflow/tensorflow/c/eager/ |
| D | gradient_checker.cc | 40 int num_dims = TF_NumDims(t); in GetDims()
61 int num_dims_out = TF_NumDims(model_out_tensor); in RunAndMaybeSum()
117 int num_dims = TF_NumDims(theta_tensor); in CalcNumericalGrad()
|
| /external/tensorflow/tensorflow/java/src/main/native/ |
| D | tensor_jni.cc | 460 const jsize num_dims = TF_NumDims(t); in Java_org_tensorflow_Tensor_shape()
476 int num_dims = TF_NumDims(t); in Java_org_tensorflow_Tensor_setValue()
494 if (TF_NumDims(t) != 0) { \
515 if (TF_NumDims(t) != 0) { in Java_org_tensorflow_Tensor_scalarBytes()
535 int num_dims = TF_NumDims(t); in Java_org_tensorflow_Tensor_readNDArray()
|
| /external/tensorflow/tensorflow/c/ |
| D | tf_tensor.h | 117 TF_CAPI_EXPORT extern int TF_NumDims(const TF_Tensor*);
|
| D | tf_tensor.cc | 151 int TF_NumDims(const TF_Tensor* t) { return t->tensor->NumDims(); } in TF_NumDims() function
163 int rank = TF_NumDims(t); in TF_TensorElementCount()
|
| D | c_api_test.cc | 140 EXPECT_EQ(2, TF_NumDims(t)); in TEST()
165 EXPECT_EQ(2, TF_NumDims(t)); in TEST()
1005 EXPECT_EQ(0, TF_NumDims(out)); // scalar in TEST()
1022 EXPECT_EQ(0, TF_NumDims(out)); // scalar in TEST()
1066 EXPECT_EQ(0, TF_NumDims(out)); // scalar in RunMinTest()
1449 EXPECT_EQ(2, TF_NumDims(out)); in TEST()
1889 ASSERT_EQ(0, TF_NumDims(t)); in ScalarFloatFromTensor()
2312 EXPECT_EQ(ndims, TF_NumDims(value)); in TEST_F()
2313 for (int i = 0; i < TF_NumDims(value); ++i) { in TEST_F()
2348 EXPECT_EQ(0, TF_NumDims(t_out)); in TEST_F()
[all …]
|
| D | while_loop_test.cc | 99 EXPECT_EQ(0, TF_NumDims(out)); in ExpectOutputValue()
|
| D | kernels_test.cc | 940 EXPECT_EQ(0, TF_NumDims(output)); in TEST_F()
986 EXPECT_EQ(num_dims, TF_NumDims(tensor)); in validate_tensor()
|
| 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()
|
| /external/tensorflow/tensorflow/cc/experimental/base/public/ |
| D | tensor.h | 129 inline int Tensor::dims() const { return TF_NumDims(tensor_.get()); } in dims()
|
| /external/tensorflow/tensorflow/c/experimental/gradients/ |
| D | grad_test_helper.cc | 85 auto num_dims_analytical = TF_NumDims(analytical_tensor); in CheckTensorValue()
|
| /external/tensorflow/tensorflow/go/ |
| D | tensor.go | 198 if ndims := int(C.TF_NumDims(c)); ndims > 0 {
|
| /external/tensorflow/tensorflow/python/lib/core/ |
| D | ndarray_tensor.cc | 310 const int ndims = TF_NumDims(tensor); in GetPyArrayDimensionsForTensor()
|