Searched refs:ConstructTensor (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes_test.cc | 1052 Tensor ConstructTensor(int data_size, const T& value = T()) { in ConstructTensor() function
1482 DataVec output_data{{"my_transpose", ConstructTensor<float>(6)}}; in TEST_F()
1569 DataVec output_data{{"my_reshape", ConstructTensor<float>(6)}}; in TEST_F()
1633 DataVec output_data{{"my_matmul", ConstructTensor<float>(2)}}; in TEST_F()
1694 {"input", ConstructTensor<CType>(num_input, CType(0))}}; in TestConvertBiasAdd()
1695 DataVec output_data{{"my_biasadd", ConstructTensor<CType>(num_input)}}; in TestConvertBiasAdd()
1793 DataVec output_data{{"my_binary", ConstructTensor<CType>(2)}}; in TestBinaryTensorOpWeightNoBroadcast()
1847 DataVec output_data{{"my_binary", ConstructTensor<CType>(4)}}; in TestBinaryTensorOpWeightWithChannelWiseBroadcast()
1882 DataVec output_data{{"my_binary", ConstructTensor<CType>(4)}}; in TestBinaryTensorOpWeightWithUniformlyBroadcast()
1933 {"input", ConstructTensor<CType>(num_inputs, CType(2))}}; in TestBinaryTensorOpWeightFallback()
[all …]
|