Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_nodes_test.cc543 Status GetTensorOrWeights(const string& name, TRT_TensorOrWeights* output) { in GetTensorOrWeights() function in tensorflow::tensorrt::convert::ConverterTest
544 return converter_->GetTensorOrWeights(name, output); in GetTensorOrWeights()
603 TF_EXPECT_OK(GetTensorOrWeights("my_op", &actual_output_1)); in TEST_F()
608 TF_EXPECT_OK(GetTensorOrWeights("my_op:1", &actual_output_2)); in TEST_F()
794 TF_EXPECT_OK(GetTensorOrWeights("my_tensor", &added_tensor)); in TEST_F()
1076 Status GetTensorOrWeights(const string& name, TRT_TensorOrWeights* output) { in GetTensorOrWeights() function in tensorflow::tensorrt::convert::OpConverterTest
1077 return converter_->GetTensorOrWeights(name, output); in GetTensorOrWeights()
1360 TF_EXPECT_OK(test->GetTensorOrWeights("my_const", &output)); in TestConvertConst()
1476 TF_EXPECT_OK(GetTensorOrWeights("my_transpose", &output)); in TEST_F()
1563 TF_EXPECT_OK(GetTensorOrWeights("my_reshape", &output)); in TEST_F()
[all …]
Dconvert_nodes.h526 Status GetTensorOrWeights(const string& name, TRT_TensorOrWeights* output);
Dconvert_nodes.cc1079 GetTensorOrWeights(output.source_tensor_name, &tensor_or_weights)); in RenameAndMarkOutputTensors()
1144 Status Converter::GetTensorOrWeights(const string& name, in GetTensorOrWeights() function in tensorflow::tensorrt::convert::Converter