Searched defs:cur_op (Results 1 – 10 of 10) sorted by relevance
| /third_party/mindspore/mindspore/lite/src/delegate/npu/pass/ |
| D | npu_fusion_pass.cc | 35 bool CheckFusion(NPUOp *cur_op) { in CheckFusion()
50 bool CheckFormatFusion(NPUOp *cur_op) { in CheckFormatFusion()
56 [](NPUOp *cur_op) { return NPUPassUtils::IsNchw2Nhwc(cur_op); }); in CheckFormatFusion()
60 [](NPUOp *cur_op) { return NPUPassUtils::IsNhwc2Nchw(cur_op); }); in CheckFormatFusion()
65 void NPUFusionPass::RemoveAndFreeOp(NPUOp *cur_op) { in RemoveAndFreeOp()
73 int NPUFusionPass::UpdatePreOps(NPUOp *cur_op) { in UpdatePreOps()
103 int NPUFusionPass::UpdatePostOps(NPUOp *cur_op) { in UpdatePostOps()
133 int UpdatePreTensors(NPUOp *cur_op) { in UpdatePreTensors()
170 bool NodeWithNhwc2nchw2nhwcOutput(NPUOp *cur_op) { in NodeWithNhwc2nchw2nhwcOutput()
182 int UpdatePostTensors(NPUOp *cur_op) { in UpdatePostTensors()
[all …]
|
| /third_party/mindspore/mindspore/lite/src/delegate/npu/ |
| D | npu_graph.cc | 99 std::vector<NPUOp *> NPUGraph::FindPreOps(NPUOp *cur_op) { in FindPreOps()
111 std::vector<NPUOp *> NPUGraph::FindNextOps(NPUOp *cur_op) { in FindNextOps()
136 auto cur_op = candidate_in_ops->front(); in FindValidSubgraphInOps() local
172 auto cur_op = op_queue.front(); in FindReadySubgraphOps() local
201 auto cur_op = bfs_ops.front(); in FindConnectedOps() local
|
| D | npu_subgraph.cc | 103 std::vector<NPUOp *> NPUSubGraph::FindPreOps(NPUOp *cur_op) { in FindPreOps()
|
| /third_party/mindspore/mindspore/lite/src/delegate/npu/op/ |
| D | arithmetic_self_npu.cc | 83 auto cur_op = reinterpret_cast<T *>(op); in SetInputs() local
|
| D | strided_slice_npu.cc | 113 int StridedSliceNPUOp::SetCast(const ge::Operator *input, const ge::Operator *cur_op, in SetCast()
|
| D | arithmetic_npu.cc | 138 auto cur_op = reinterpret_cast<T *>(op); in SetInputs() local
|
| D | scale_npu.cc | 124 … ScaleNPUOp::ConvertScaleToMul(const std::vector<ge::Operator *> &npu_inputs, ge::Operator *cur_op, in ConvertScaleToMul()
|
| /third_party/mindspore/mindspore/lite/src/delegate/ |
| D | delegate_utils.h | 144 std::vector<T *> FindPreOps(T *cur_op, std::vector<T *> all_ops) { in FindPreOps()
157 std::vector<T *> FindNextOps(T *cur_op, std::vector<T *> all_ops) { in FindNextOps()
|
| /third_party/ffmpeg/libavfilter/dnn/ |
| D | dnn_backend_tf.c | 317 … DNNReturnType add_conv_layer(TFModel *tf_model, TF_Operation *transpose_op, TF_Operation **cur_op, in add_conv_layer()
442 static DNNReturnType add_depth_to_space_layer(TFModel *tf_model, TF_Operation **cur_op, in add_depth_to_space_layer()
466 static DNNReturnType add_pad_layer(TFModel *tf_model, TF_Operation **cur_op, in add_pad_layer()
521 static DNNReturnType add_maximum_layer(TFModel *tf_model, TF_Operation **cur_op, in add_maximum_layer()
|
| /third_party/mindspore/mindspore/lite/src/delegate/tensorrt/ |
| D | tensorrt_subgraph.cc | 207 for (auto cur_op : all_ops_) { in BuildTensorRTGraph() local
484 ITensorHelper TensorRTSubGraph::FindTensorRTInputs(TensorRTOp *cur_op, const mindspore::MSTensor &i… in FindTensorRTInputs()
|