| /external/tensorflow/tensorflow/lite/micro/kernels/ |
| D | reshape_test.cc | 37 const size_t expected_output_len, const int* expected_dims, in ValidateReshapeGoldens() argument
61 TF_LITE_MICRO_EXPECT_EQ(expected_dims[i], output_tensor->dims->data[i]); in ValidateReshapeGoldens()
69 const int* expected_dims, in TestReshapeWithShape() argument
85 expected_output, expected_output_len, expected_dims, in TestReshapeWithShape()
95 const int* expected_dims, in TestReshapeWithoutShape() argument
111 expected_output, expected_output_len, expected_dims, in TestReshapeWithoutShape()
119 const int* expected_dims, const size_t expected_dims_len, in TestReshape() argument
129 expected_output, expected_output_len, expected_dims, in TestReshape()
139 const int* expected_dims, in TestReshapeQuantized() argument
152 expected_output, expected_output_len, expected_dims, in TestReshapeQuantized()
[all …]
|
| /external/tensorflow/tensorflow/python/ops/ragged/ |
| D | ragged_tensor_shape_test.py | 313 dict(x_dims=[], y_dims=[], expected_dims=[]),
314 dict(x_dims=[], y_dims=[2], expected_dims=[2]),
315 dict(x_dims=[], y_dims=[2, 3], expected_dims=[2, 3]),
319 expected_dims=[2, (2, 3), (5, 7, 2, 0, 9)]),
321 dict(x_dims=[3], y_dims=[4, 2, 3], expected_dims=[4, 2, 3]),
322 dict(x_dims=[1], y_dims=[4, 2, 3], expected_dims=[4, 2, 3]),
323 dict(x_dims=[3], y_dims=[4, 2, 1], expected_dims=[4, 2, 3]),
327 expected_dims=[3, (2, 3, 1), 3]),
328 dict(x_dims=[1], y_dims=[3, (2, 1, 3)], expected_dims=[3, (2, 1, 3)]),
332 expected_dims=[3, (2, 1, 3), 8]),
[all …]
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | shape_inference_testutil.cc | 143 auto expected_dims = str_util::Split(expected, ','); in InferShapes() local
145 return Unknown(err_prefix, " expected rank ", expected_dims.size(), in InferShapes()
148 if (c.Rank(out) != expected_dims.size()) { in InferShapes()
149 return Unknown(err_prefix, " expected rank ", expected_dims.size(), in InferShapes()
152 for (int j = 0; j < expected_dims.size(); ++j) { in InferShapes()
154 StringPiece expected_dim(expected_dims[j]); in InferShapes()
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | common_shapes_test.py | 146 expected_dims = expected.dims
147 if expected_dims is None:
150 self.assertIsNone(expected_dims)
152 self.assertEqual(len(expected_dims), len(actual_dims))
153 for expected_dim, actual_dim in zip(expected_dims, actual_dims):
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | transpose_util_test.cc | 28 const gtl::ArraySlice<int64>& expected_dims) { in TestDimensionReduction() argument
36 EXPECT_EQ(computed_dims, expected_dims); in TestDimensionReduction()
|
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_concat_op.cc | 485 size_t expected_dims = expected_shape.dims(); in Compute() local
487 if (concat_dim < 0) concat_dim = expected_dims + concat_dim; in Compute()
495 context, s_dims == expected_dims, in Compute()
499 s_dims, " vs. expected rank = ", expected_dims)); in Compute()
501 for (int d = 0; d < expected_dims; ++d) { in Compute()
537 if (expected_dims == 2) invoke_eigen = true; in Compute()
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes_test.cc | 293 const std::vector<int>& expected_dims, in ValidateWeights() argument
295 ExpectTrtDimsEqualsArray(expected_dims, weights.shape_); in ValidateWeights()
2021 const std::vector<int>& expected_dims, in TestConvertConst()
2046 ValidateWeights(output.weights(), expected_dims, expected_value); in TestConvertConst()
|