Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
H A Dop_converter_test.cc83 ->AddInputTensor("input1", nvinfer1::DataType::kFLOAT, in TEST()
95 ->AddInputTensor("input2", nvinfer1::DataType::kFLOAT, in TEST()
H A Dconvert_nodes.h280 Status AddInputTensor(const string& name, nvinfer1::DataType dtype,
H A Dconvert_nodes_test.cc589 TF_ASSERT_OK(converter_->AddInputTensor( in TEST_F()
625 TF_EXPECT_OK(converter_->AddInputTensor("input", nvinfer1::DataType::kFLOAT, in TEST_F()
627 TF_EXPECT_OK(converter_->AddInputTensor("input:1", nvinfer1::DataType::kINT32, in TEST_F()
629 TF_EXPECT_OK(converter_->AddInputTensor( in TEST_F()
677 TF_EXPECT_OK(converter_->AddInputTensor( in TEST_F()
1367 Status status = converter_->AddInputTensor( in AddTestTensorWithTFDims()
H A Dconvert_nodes.cc1115 Status Converter::AddInputTensor(const string& name, nvinfer1::DataType dtype, in AddInputTensor() function in tensorflow::tensorrt::convert::Converter
5940 TF_RETURN_IF_ERROR(converter->AddInputTensor(node_name, trt_dtype, in ConvertGraphDefToEngine()