Searched refs:fake_tensor (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes_test.cc | 1035 FakeITensor fake_tensor; in TEST_F() local
1036 TRT_TensorOrWeights tensor(&fake_tensor); in TEST_F()
1074 FakeITensor fake_tensor; in TEST_F() local
1076 converter_->ProvideQuantizationRange(&fake_tensor, 0.0f, 6.0f); in TEST_F()
1077 EXPECT_EQ(6.0f, quantization_ranges()[&fake_tensor]); in TEST_F()
1078 converter_->ProvideQuantizationRange(&fake_tensor, 1.0f, 6.0f); in TEST_F()
1079 EXPECT_EQ(6.0f, quantization_ranges()[&fake_tensor]); in TEST_F()
1080 converter_->ProvideQuantizationRange(&fake_tensor, -8.0f, 6.0f); in TEST_F()
1081 EXPECT_EQ(8.0f, quantization_ranges()[&fake_tensor]); in TEST_F()
1082 converter_->ProvideQuantizationRange(&fake_tensor, -8.123f, -6.123f); in TEST_F()
[all …]
|