| /external/tensorflow/tensorflow/python/compiler/tensorrt/test/ |
| D | lru_cache_test.py | 40 expected_output_dims = [[[1, 10, 10, 1]], [[2, 10, 10, 1]], [[4, 10, 10,
66 expected_output_dims=expected_output_dims)
|
| D | base_test.py | 72 expected_output_dims=[[[100, 6, 6, 6]]])
129 expected_output_dims=[[[100, 12, 12, 6]]])
181 expected_output_dims=[[input_dims]])
234 expected_output_dims=[[input_dims]])
267 expected_output_dims=[[input_dims]])
301 expected_output_dims=[[input_dims]])
350 expected_output_dims=[[input_dims]])
|
| D | dynamic_input_shapes_test.py | 44 expected_output_dims = input_dims
78 expected_output_dims=expected_output_dims)
|
| D | conv2d_test.py | 109 expected_output_dims=[[[13, 5, 7, 11]]])
134 expected_output_dims=[[[13, 7, 11, 5]]])
184 expected_output_dims=[[[n, num_filters, h, w]]])
225 expected_output_dims=[[[n, num_filters, h * 2, w * 2]]])
|
| D | topk_test.py | 50 expected_output_dims=[[[100, k], [100, k]]])
81 expected_output_dims=[[[100, k], [100, 1, k]]])
|
| D | reshape_transpose_test.py | 77 expected_output_dims=[[input_dims]])
134 expected_output_dims=[[[24, 100, 2, 24]]])
|
| D | identity_output_test.py | 66 expected_output_dims=[[[100, 4]] * 3])
|
| D | binary_tensor_weight_broadcast_test.py | 68 expected_output_dims=[[[5, 23040]]])
|
| D | memory_alignment_test.py | 67 expected_output_dims=[[[2, 15, 15, 10]]])
|
| D | neighboring_engine_test.py | 64 expected_output_dims=[[[2, 4, 5, 4]]])
|
| D | int32_test.py | 57 expected_output_dims=[[[100, 10]]])
|
| D | vgg_block_test.py | 66 expected_output_dims=[[[5, 2, 2, 6]]])
|
| D | const_broadcast_test.py | 63 expected_output_dims=[[[5, 12, 12, 1]]])
|
| D | vgg_block_nchw_test.py | 75 expected_output_dims=[[[5, 6, 2, 2]]])
|
| D | multi_connection_neighbor_engine_test.py | 80 expected_output_dims=[[[2, 4, 5, 4]]])
|
| D | rank_two_test.py | 68 expected_output_dims=[[input_dims[1]]])
|
| D | batch_matmul_test.py | 76 expected_output_dims=[[[12, 5, 8, 7]]])
|
| D | quantization_test.py | 65 expected_output_dims=[[[8, 1]]])
|
| D | concatenation_test.py | 78 expected_output_dims=[[[2, 126]]])
|
| D | unary_test.py | 105 expected_output_dims=[[[12, 5, 8, 12]]])
|
| D | biasadd_matmul_test.py | 116 expected_output_dims=[[[4, 6680]]])
|
| D | tf_trt_integration_test_base.py | 286 output_len = len(params.expected_output_dims[shape_index])
290 list(params.expected_output_dims[shape_index][i]),
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes_test.cc | 1919 std::vector<int> expected_output_dims = input_dims; in TestBinaryTensorOpWeightFallback() local
1920 for (int i = expected_output_dims.size() - 1, j = weights_dims.size() - 1; in TestBinaryTensorOpWeightFallback()
1922 if (expected_output_dims[i] == 1) { in TestBinaryTensorOpWeightFallback()
1923 expected_output_dims[i] = weights_dims[j]; in TestBinaryTensorOpWeightFallback()
1926 ExpectTrtDimsEqualsArray(expected_output_dims, in TestBinaryTensorOpWeightFallback()
1931 TrtDimsNumElements(GetTestDims(expected_output_dims)); in TestBinaryTensorOpWeightFallback()
2457 std::vector<int> expected_output_dims; in TEST_F() member
2476 ExpectTrtDimsEqualsArray(ok_params[i].expected_output_dims, in TEST_F()
2569 std::vector<int> expected_output_dims; in TEST_F() member
2594 ExpectTrtDimsEqualsArray(ok_params[i].expected_output_dims, in TEST_F()
[all …]
|