Searched defs:slice_input (Results 1 – 12 of 12) sorted by relevance
| /third_party/mindspore/mindspore-src/source/mindspore/lite/src/extendrt/delegate/tensorrt/op/ |
| D | strideslice_tensorrt.cc | 92 …sor *StrideSliceTensorRT::GetDynamicSliceSize(TensorRTContext *ctx, nvinfer1::ITensor *slice_input, in GetDynamicSliceSize()
204 int StrideSliceTensorRT::ComputeDimsMulti(TensorRTContext *ctx, ITensorHelper *slice_input, const T… in ComputeDimsMulti()
245 int StrideSliceTensorRT::ComputeDimsSingle(TensorRTContext *ctx, ITensorHelper *slice_input, const … in ComputeDimsSingle()
292 int StrideSliceTensorRT::ComputeDims(TensorRTContext *ctx, ITensorHelper *slice_input, const Tensor… in ComputeDims()
318 int StrideSliceTensorRT::ComputeSliceDims(TensorRTContext *ctx, ITensorHelper *slice_input) { in ComputeSliceDims()
|
| D | slicefusion_tensorrt.cc | 89 ITensorHelper slice_input; in AddInnerOp() local
|
| D | slice_tensorrt.cc | 298 ITensorHelper slice_input; in AddInnerOp() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/minddata/dataset/kernels/data/ |
| D | slice_op.h | 33 explicit SliceOp(std::vector<SliceOption> slice_input) : slice_options_(std::move(slice_input)) {} in SliceOp()
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/src/extendrt/delegate/tensorrt/optimizer/ |
| D | tensorrt_optimizer.cc | 138 auto slice_input = common::AnfAlgo::GetInputNode(slice_node, kIndex0); in GetMatmulFactor() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/pass/ |
| D | slice_activation_in_cell_share_recompute.cc | 35 auto slice_input = parallel::CreateInput(slice_op, node, parallel::STRIDEDSLICE); in CreateStridedSliceCNode() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/minddata/dataset/api/python/bindings/dataset/kernels/ir/ |
| D | bindings.cc | 174 .def(py::init([](std::vector<SliceOption> slice_input) { in __anon2fd259f71602()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/optimizer/ |
| D | slice_activation_in_recompute.cc | 41 auto slice_input = parallel::CreateInput(slice_op, node, parallel::STRIDEDSLICE); in CreateStridedSliceCNode() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/minddata/dataset/api/ |
| D | transforms.cc | 260 Slice::Slice(const std::vector<SliceOption> &slice_input) : data_(std::make_shared<Data>(slice_inpu… in Slice()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/minddata/dataset/kernels/ir/data/ |
| D | transforms_ir.cc | 440 SliceOperation::SliceOperation(const std::vector<SliceOption> &slice_input) : slice_input_(slice_in… in SliceOperation()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/optimizer/ |
| D | neighbor_exchange_v2_fusion.cc | 151 CNodePtr CreateSliceNode(const FuncGraphPtr &graph, const std::vector<AnfNodePtr> &slice_input, in CreateSliceNode()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/ |
| D | step_parallel_utils.cc | 2994 auto slice_input = need_replace_op_lists[kIndex0][col]->input(kIndex1); in MergeOpBeforeInterleaveSlice() local
|