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.h470 Status AddInputTensor(const string& name, nvinfer1::DataType dtype,
Dconvert_nodes_test.cc805 TF_EXPECT_OK(converter_->AddInputTensor( in TEST_F()
837 TF_EXPECT_OK(converter_->AddInputTensor("input", nvinfer1::DataType::kFLOAT, in TEST_F()
839 TF_EXPECT_OK(converter_->AddInputTensor("input:1", nvinfer1::DataType::kINT32, in TEST_F()
841 TF_EXPECT_OK(converter_->AddInputTensor( in TEST_F()
888 TF_EXPECT_OK(converter_->AddInputTensor( in TEST_F()
1549 converter_->AddInputTensor(name, trt_type, trt_dims, batch_size); in AddTestTensorWithTFDims()
Dconvert_nodes.cc1332 Status Converter::AddInputTensor(const string& name, nvinfer1::DataType dtype, in AddInputTensor() function in tensorflow::tensorrt::convert::Converter
6386 TF_RETURN_IF_ERROR(converter->AddInputTensor(node_name, trt_dtype, in ConvertGraphDefToEngine()