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.h468 Status AddInputTensor(const string& name, nvinfer1::DataType dtype,
Dconvert_nodes_test.cc725 TF_EXPECT_OK(converter_->AddInputTensor( in TEST_F()
759 TF_EXPECT_OK(converter_->AddInputTensor("input", nvinfer1::DataType::kFLOAT, in TEST_F()
761 TF_EXPECT_OK(converter_->AddInputTensor("input:1", nvinfer1::DataType::kINT32, in TEST_F()
763 TF_EXPECT_OK(converter_->AddInputTensor( in TEST_F()
811 TF_EXPECT_OK(converter_->AddInputTensor( in TEST_F()
1458 converter_->AddInputTensor(name, trt_type, trt_dims, batch_size); in AddTestTensorWithTFDims()
Dconvert_nodes.cc1355 Status Converter::AddInputTensor(const string& name, nvinfer1::DataType dtype, in AddInputTensor() function in tensorflow::tensorrt::convert::Converter
6863 TF_RETURN_IF_ERROR(converter->AddInputTensor(node_name, trt_dtype, in ConvertGraphDefToEngine()