Searched refs:FindTensorRTInputs (Results 1 – 2 of 2) sorted by relevance
73 ITensorHelper FindTensorRTInputs(TensorRTOp *cur_op, const mindspore::MSTensor &in_tensor);
220 ITensorHelper trt_tensor = FindTensorRTInputs(cur_op, in_tensor); in BuildTensorRTGraph()484 ITensorHelper TensorRTSubGraph::FindTensorRTInputs(TensorRTOp *cur_op, const mindspore::MSTensor &i… in FindTensorRTInputs() function in mindspore::lite::TensorRTSubGraph