Home
last modified time | relevance | path

Searched refs:AddInputTensor (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_nodes.h447 Status AddInputTensor(const string& name, nvinfer1::DataType dtype,
Dconvert_nodes_test.cc591 TF_EXPECT_OK(converter_->AddInputTensor( in TEST_F()
621 TF_EXPECT_OK(converter_->AddInputTensor("input", nvinfer1::DataType::kFLOAT, in TEST_F()
623 TF_EXPECT_OK(converter_->AddInputTensor("input:1", nvinfer1::DataType::kINT32, in TEST_F()
625 TF_EXPECT_OK(converter_->AddInputTensor( in TEST_F()
671 TF_EXPECT_OK(converter_->AddInputTensor( in TEST_F()
1208 converter_->AddInputTensor(name, trt_dtype, trt_dims, batch_size)); in AddTestTensor()
Dconvert_nodes.cc1051 Status Converter::AddInputTensor(const string& name, nvinfer1::DataType dtype, in AddInputTensor() function in tensorflow::tensorrt::convert::Converter
4128 converter.AddInputTensor(node_name, trt_dtype, trt_dims, batch_size)); in ConvertGraphDefToEngine()