Home
last modified time | relevance | path

Searched defs:StridedSlice (Results 1 – 14 of 14) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/graph_kernel/converter/expanders/
Dstrided_slice.cc22 class StridedSlice : public OpDesc { class
24 StridedSlice() {} in StridedSlice() function in mindspore::graphkernel::expanders::StridedSlice
/third_party/mindspore/mindspore-src/source/tests/st/ops/cpu/
Dtest_stridedslice_op.py31 class StridedSlice(nn.Cell): class
Dtest_slice_op.py176 class StridedSlice(nn.Cell): class
/third_party/mindspore/mindspore-src/source/tests/st/dynamic_shape/
Dtest_dynamic_stridedslice.py26 class StridedSlice(nn.Cell): class
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/graph_kernel/expander/base/
Dmeta_op.h55 StridedSlice, enumerator
Dir_builder.h155 inline NodePtr StridedSlice(const NodePtr &input, const NodePtr &begin, const NodePtr &end, in StridedSlice() function
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/graph_kernel/model/
Dgraph_builder.cc73 NodePtr GraphBuilder::StridedSlice(const NodePtr &input, const std::vector<int64_t> &begin, in StridedSlice() function in mindspore::graphkernel::inner::GraphBuilder
/third_party/mindspore/mindspore-src/source/mindspore/core/ops/symbol_ops_impl/
Dstrided_slice.cc26 class MS_CORE_API StridedSlice : public InferShapeOp { class
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/expander/bprop/
Dbprop_irbuilder.h98 …NodePtr StridedSlice(const NodePtr &dout, const NodePtr &begin, const NodePtr &end, const NodePtr … in StridedSlice() function
Dbprop_irbuilder.cc145 NodePtr BpropBuilder::StridedSlice(const NodePtr &x, const std::map<int64_t, std::vector<int64_t>> … in StridedSlice() function in mindspore::expander::bprop::BpropBuilder
/third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/_extends/
Dbuiltin_operations.py198 def StridedSlice(x, begin, end, stride): function
/third_party/mindspore/mindspore-src/source/mindspore/lite/test/ut/src/runtime/kernel/arm/common/
Dstrided_slice_tests.cc52 TEST_F(TestStridedSlice, StridedSlice) { in TEST_F() argument
/third_party/mindspore/mindspore-src/source/mindspore/core/ops/auto_generate/
Dgen_lite_ops.h2597 StridedSlice() : BaseOperator(kNameStridedSlice) {} in StridedSlice() function
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/cuda_impl/cuda_ops/
Dslice_impl.cu290 cudaError_t StridedSlice(const std::vector<size_t> &input_shape, const std::vector<int64_t> &begin, in StridedSlice() function