Searched refs:ConstructTensor (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes_test.cc | 1437 Tensor ConstructTensor(int data_size, const T& value = T()) { in ConstructTensor() function in tensorflow::tensorrt::convert::OpConverterTest
1444 Tensor ConstructTensor(int data_size, const T& value, DataType tf_type) { in ConstructTensor() function in tensorflow::tensorrt::convert::OpConverterTest
1884 out_name, ConstructTensor(shape.num_elements(), 0, out_tf_type)};
2413 {"my_reshape", ConstructTensor<float>(input_vec.size())}}; in TEST_F()
2517 DataVec output_data{{"my_matmul", test->ConstructTensor<float>(2)}}; in TestMatMulHelper()
2544 DataVec output_data{{"my_matmul", test->ConstructTensor<float>(2)}}; in TestMatMulHelper()
2687 DataVec output_data{{"my_matmul", ConstructTensor<float>(4)}}; in TEST_F()
4196 ConstructTensor<float>(ok_params[i].expected_output.size())}}; in TEST_F()
4335 DataVec output_data{{"my_slice", ConstructTensor<float>( in TEST_F()
4685 ConstructTensor<float>(ok_params[i].expected_output.size())}}; in TEST_F()
[all …]
|