Searched defs:expected_dims (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/lite/micro/kernels/ |
| D | reshape_test.cc | 37 const size_t expected_output_len, const int* expected_dims, in ValidateReshapeGoldens()
69 const int* expected_dims, in TestReshapeWithShape()
95 const int* expected_dims, in TestReshapeWithoutShape()
119 const int* expected_dims, const size_t expected_dims_len, in TestReshape()
139 const int* expected_dims, in TestReshapeQuantized()
224 const int expected_dims[] = {}; in TF_LITE_MICRO_TEST() local
351 const int expected_dims[] = {}; in TF_LITE_MICRO_TEST() local
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | transpose_util_test.cc | 28 const gtl::ArraySlice<int64>& expected_dims) { in TestDimensionReduction()
|
| /external/tensorflow/tensorflow/python/ops/ragged/ |
| D | ragged_tensor_shape_test.py | 365 def testBroadcastDynamicShape(self, x_dims, y_dims, expected_dims): argument
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | shape_inference_testutil.cc | 143 auto expected_dims = str_util::Split(expected, ','); in InferShapes() local
|
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_concat_op.cc | 485 size_t expected_dims = expected_shape.dims(); in Compute() local
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes_test.cc | 293 const std::vector<int>& expected_dims, in ValidateWeights()
2022 const std::vector<OutputCType>& expected_value) { in TestConvertConst()
|