Searched refs:FakeITensor (Results 1 – 1 of 1) sorted by relevance
206 class FakeITensor : public nvinfer1::ITensor { class208 FakeITensor() : dynamic_range_(0.0f) {} in FakeITensor() function in tensorflow::tensorrt::convert::FakeITensor210 FakeITensor(const nvinfer1::Dims& dims) : dims_(dims), dynamic_range_(0.0f) {} in FakeITensor() function in tensorflow::tensorrt::convert::FakeITensor212 FakeITensor(const std::vector<int>& dims) in FakeITensor() function in tensorflow::tensorrt::convert::FakeITensor339 FakeITensor itensor(dims); in TEST()580 FakeITensor output_tensors[2]; in TEST_F()786 FakeITensor fake_tensor; in TEST_F()824 FakeITensor fake_tensor; in TEST_F()841 FakeITensor input, infer_1, infer_2, infer_3; in TEST_F()842 FakeITensor not_infer; in TEST_F()[all …]