Searched refs:expected_output_dims (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/python/compiler/tensorrt/test/ |
| D | lru_cache_test.py | 46 expected_output_dims = [[[1, 10, 10, 2]], [[2, 10, 10, 2]], [[4, 10, 10,
58 expected_output_dims=expected_output_dims)
|
| D | dynamic_input_shapes_test.py | 68 expected_output_dims = input_dims
81 expected_output_dims=expected_output_dims)
|
| D | tf_trt_integration_test_base.py | 262 expected_output_dims=[output_shapes] + extra_outputs)
377 params.expected_output_dims, inputs_data):
399 for expected_shapes, current_input_data in zip(params.expected_output_dims,
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes_test.cc | 1735 std::vector<int> expected_output_dims; member
1752 if (!p.expected_output_dims.empty()) { in operator <<()
1753 os << ", exp_out_dims" << p.expected_output_dims; in operator <<()
1869 const std::vector<std::vector<int>>& expected_output_dims, in BuildAndRun() argument
1874 const int n_output = expected_output_dims.size();
1879 TensorShapeUtils::MakeShape(expected_output_dims[i], &shape));
1898 TensorShapeUtils::MakeShape(expected_output_dims[i], &shape));
1912 const std::vector<std::vector<int>>& expected_output_dims, in TestOpConverterMultiOut() argument
1918 name.c_str(), expected_output_dims);
1920 BuildAndRun(name, expected_output_dims, expected_runtime_status, matcher,
[all …]
|
| /external/tensorflow/tensorflow/lite/delegates/xnnpack/ |
| D | xnnpack_delegate.cc | 1993 const int expected_output_dims = reducer_params->keep_dims ? 4 : 2; in VisitMeanNode() local
1995 expected_output_dims, in VisitMeanNode()
|