Searched refs:kPrimStridedSlice (Results 1 – 12 of 12) sorted by relevance
| /third_party/mindspore/mindspore/core/ops/ |
| D | strided_slice.h | 34 StridedSlice() : PrimitiveC(prim::kPrimStridedSlice->name()) { in StridedSlice()
|
| /third_party/mindspore/mindspore/lite/tools/converter/parser/ |
| D | tf_bidirection_gru_cf_fusion.cc | 87 …auto strided_slice = VectorRef({GetPrim(prim::kPrimStridedSlice), fw_shape1, std::make_shared<SeqV… in DefineBidirectionRnnPattern()
110 …auto fw_unstack_strided_slice = VectorRef({GetPrim(prim::kPrimStridedSlice), fw_shape, std::make_s… in DefineBidirectionRnnPattern()
|
| /third_party/mindspore/mindspore/lite/tools/optimizer/graph/ |
| D | transpose_strategy.cc | 438 …imitiveType(cnode, prim::kPrimSliceFusion) || CheckPrimitiveType(cnode, prim::kPrimStridedSlice) || in CanChangeOpAxis()
448 if (CheckPrimitiveType(cnode, prim::kPrimStridedSlice) && cnode->size() != kOnnxStridedSlice) { in CanChangeOpAxis()
472 …{prim::kPrimSliceFusion->name(), ChangeOpSlice}, {prim::kPrimStridedSlice->name(), ChangeOpStrideS… in ChangeOpAxis()
|
| /third_party/mindspore/mindspore/lite/tools/optimizer/fusion/ |
| D | tflite_rel_pos_multi_head_attention_fusion.cc | 361 …to is_strided_slice1 = std::make_shared<CondVar>(std::bind(IsOpType, p1, prim::kPrimStridedSlice)); in DefineRelativeShiftPattern()
371 …to is_strided_slice2 = std::make_shared<CondVar>(std::bind(IsOpType, p1, prim::kPrimStridedSlice)); in DefineRelativeShiftPattern()
|
| D | tf_bidirection_gru_fusion.cc | 141 …auto is_strided_slice = std::make_shared<CondVar>(std::bind(IsOpType, p1, prim::kPrimStridedSlice)… in DefineFowardPattern()
211 …auto is_strided_slice = std::make_shared<CondVar>(std::bind(IsOpType, p1, prim::kPrimStridedSlice)… in DefinebackwardPattern()
|
| /third_party/mindspore/mindspore/core/abstract/ |
| D | primitive_infer_map.cc | 246 {prim::kPrimStridedSlice, {ops::StridedSliceInfer, nullptr, true}}, in GetPrimitiveToBackendEvalImplMap()
|
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/graph_util/ |
| D | pipeline_split_utils.cc | 87 if (!IsPrimitiveCNode(node, prim::kPrimStridedSlice)) { in SetStridedSliceStrategy()
618 if (stage_id != 0 && IsPrimitiveCNode(node, prim::kPrimStridedSlice)) { in GetBorderNode()
|
| /third_party/mindspore/mindspore/lite/tools/converter/parser/onnx/ |
| D | onnx_inputs_adjust.cc | 389 } else if (opt::CheckPrimitiveType(node, prim::kPrimStridedSlice)) { in Adjust()
|
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/pipeline_transformer/ |
| D | pipeline_transformer.cc | 77 if (!IsPrimitiveCNode(node, prim::kPrimStridedSlice)) { in SetMicroBatch()
161 if (!IsPrimitiveCNode(node_first, prim::kPrimStridedSlice)) { in LabelMicroBatch()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/graph_kernel/ |
| D | graph_kernel_cluster.cc | 95 {kGPUDevice, OpLevel_0, prim::kPrimStridedSlice}, in GetClusterableOpList()
|
| /third_party/mindspore/mindspore/ccsrc/transform/express_ir/ |
| D | onnx_exporter.cc | 855 auto name = prim::kPrimStridedSlice->name(); in ExportPrimStridedSlice()
1584 if (node->IsApply(prim::kPrimStridedSlice)) { in ExportCNode()
|
| /third_party/mindspore/mindspore/core/base/ |
| D | core_ops.h | 185 inline const PrimitivePtr kPrimStridedSlice = std::make_shared<Primitive>(kStridedSlice); variable
|