Home
last modified time | relevance | path

Searched defs:trt_tensor (Results 1 – 3 of 3) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/lite/src/extendrt/delegate/tensorrt/op/
Dcast_tensorrt.cc54 auto trt_tensor = input(ctx, 0).trt_tensor_; in AddInnerOp() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/extendrt/delegate/tensorrt/
Dtensorrt_subgraph.cc370 …nvinfer1::ITensor *trt_tensor = SetTensorRTNetworkInput(in_tensor, GetInputIndexByName(in_tensor.N… in BuildTensorRTGraph() local
382 ITensorHelper trt_tensor = FindTensorRTInputs(cur_op, in_tensor); in BuildTensorRTGraph() local
440 …nvinfer1::ITensor *trt_tensor = SetTensorRTNetworkInput(*input_it, GetInputIndexByName(input_it->N… in MarkOutputs() local
Dtensorrt_utils.cc513 nvinfer1::ITensor *TRTTensorCast(TensorRTContext *ctx, nvinfer1::ITensor *trt_tensor, nvinfer1::Dat… in TRTTensorCast()
671 std::string GetTensorFormat(nvinfer1::ITensor *trt_tensor, mindspore::Format format, bool is_same) { in GetTensorFormat()
692 std::string GetTensorFormat(nvinfer1::ITensor *trt_tensor) { return GetTensorFormat(trt_tensor, For… in GetTensorFormat()