Searched refs:tensor_or_weights (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes.cc | 896 TRT_TensorOrWeights* tensor_or_weights) { in ConvertToTensorOrWeights() argument
909 return ConvertConstToWeights(node_def, inputs, tensor_or_weights); in ConvertToTensorOrWeights()
931 *tensor_or_weights = TRT_TensorOrWeights(trt_dtype, trt_dims, batch_size); in ConvertToTensorOrWeights()
962 TRT_TensorOrWeights tensor_or_weights; in ValidateNode() local
964 *pair.first, pair.second, graph_properties, &tensor_or_weights); in ValidateNode()
970 inputs.push_back(tensor_or_weights); in ValidateNode()
1077 TRT_TensorOrWeights tensor_or_weights; in RenameAndMarkOutputTensors() local
1079 GetTensorOrWeights(output.source_tensor_name, &tensor_or_weights)); in RenameAndMarkOutputTensors()
1080 if (!tensor_or_weights.is_tensor()) { in RenameAndMarkOutputTensors()
1084 nvinfer1::ITensor* tensor = tensor_or_weights.tensor(); in RenameAndMarkOutputTensors()
|
| D | convert_nodes.h | 404 TRT_TensorOrWeights* tensor_or_weights);
|
| D | convert_nodes_test.cc | 404 TRT_TensorOrWeights* tensor_or_weights) { in ConvertToTensorOrWeights() argument
406 node_def, output_port, graph_properties, tensor_or_weights); in ConvertToTensorOrWeights()
|