Searched refs:kPrimSliceFusion (Results 1 – 5 of 5) sorted by relevance
| /third_party/mindspore/mindspore/lite/tools/optimizer/fusion/ |
| D | batchmatmul_fusion.cc | 192 if (!CheckPrimitiveType(left_slice_cnode, prim::kPrimSliceFusion)) { in Process()
201 …if (left_slice_cnode == nullptr || !CheckPrimitiveType(left_slice_cnode, prim::kPrimSliceFusion)) { in Process()
|
| /third_party/mindspore/mindspore/lite/tools/converter/parser/tflite/ |
| D | tflite_inputs_adjust.cc | 198 if (opt::CheckPrimitiveType(node, prim::kPrimSliceFusion)) { in Run()
|
| /third_party/mindspore/mindspore/lite/tools/optimizer/graph/ |
| D | slice_prepose_pass.cc | 415 if (!CheckPrimitiveType(cnode, prim::kPrimSliceFusion)) { in SiblingsAreSameSlice()
1437 if (!CheckPrimitiveType(first_slice, prim::kPrimSliceFusion)) { in MergeParallelSlice()
1454 if (!CheckPrimitiveType(slice, prim::kPrimSliceFusion)) { in MergeParallelSlice()
1491 } else if (CheckPrimitiveType(preceed_cnode, prim::kPrimSliceFusion)) { in DoPrepose()
1509 !CheckPrimitiveType(node, prim::kPrimSliceFusion)) { in Run()
|
| D | transpose_strategy.cc | 438 …if (CheckPrimitiveType(cnode, prim::kPrimSliceFusion) || CheckPrimitiveType(cnode, prim::kPrimStri… in CanChangeOpAxis()
472 …{prim::kPrimSliceFusion->name(), ChangeOpSlice}, {prim::kPrimStridedSlice->name(), ChangeOpStrideS… in ChangeOpAxis()
|
| /third_party/mindspore/mindspore/core/base/ |
| D | core_ops.h | 210 inline const PrimitivePtr kPrimSliceFusion = std::make_shared<Primitive>("SliceFusion"); variable
|