Searched refs:fake_tensor (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes_test.cc | 786 FakeITensor fake_tensor; in TEST_F() local
787 TRT_TensorOrWeights tensor(&fake_tensor); in TEST_F()
824 FakeITensor fake_tensor; in TEST_F() local
826 converter_->ProvideQuantizationRange(&fake_tensor, 0.0f, 6.0f); in TEST_F()
827 EXPECT_EQ(6.0f, quantization_ranges()[&fake_tensor]); in TEST_F()
828 converter_->ProvideQuantizationRange(&fake_tensor, 1.0f, 6.0f); in TEST_F()
829 EXPECT_EQ(6.0f, quantization_ranges()[&fake_tensor]); in TEST_F()
830 converter_->ProvideQuantizationRange(&fake_tensor, -8.0f, 6.0f); in TEST_F()
831 EXPECT_EQ(8.0f, quantization_ranges()[&fake_tensor]); in TEST_F()
832 converter_->ProvideQuantizationRange(&fake_tensor, -8.123f, -6.123f); in TEST_F()
[all …]
|