Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_nodes.cc4778 std::vector<nvinfer1::ITensor*> expanded_tensors; in ConvertPack() local
4795 expanded_tensors.push_back(expanded_tensor); in ConvertPack()
4803 params->outputs->push_back(TRT_TensorOrWeights(expanded_tensors[0])); in ConvertPack()
4810 const_cast<nvinfer1::ITensor**>(expanded_tensors.data()), in ConvertPack()
4811 expanded_tensors.size()); in ConvertPack()