| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp32_grad/ |
| D | strided_slice_grad.cc | 77 int32_t real_index = param_->in_shape_length_ - 1; in FillEmptyDims() local
79 if (real_index >= 0) { in FillEmptyDims()
80 param_->begins_[i] = begins[real_index]; in FillEmptyDims()
81 param_->ends_[i] = ends[real_index]; in FillEmptyDims()
82 param_->strides_[i] = strides[real_index]; in FillEmptyDims()
83 param_->in_shape_[i] = input_shape[real_index--]; in FillEmptyDims()
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp16_grad/ |
| D | strided_slice_fp16_grad.cc | 76 int32_t real_index = param_->in_shape_length_ - 1; in FillEmptyDims() local
78 if (real_index >= 0) { in FillEmptyDims()
79 param_->begins_[i] = begins[real_index]; in FillEmptyDims()
80 param_->ends_[i] = ends[real_index]; in FillEmptyDims()
81 param_->strides_[i] = strides[real_index]; in FillEmptyDims()
82 param_->in_shape_[i] = input_shape[real_index--]; in FillEmptyDims()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/base/ |
| D | slice_base.c | 29 int32_t real_index = param->param_length_ - 1; in PadSliceParameterTo8D() local
31 if (real_index >= 0) { in PadSliceParameterTo8D()
32 param->begin_[i] = begin[real_index]; in PadSliceParameterTo8D()
33 param->end_[i] = end[real_index]; in PadSliceParameterTo8D()
34 param->size_[i] = slice_size[real_index]; in PadSliceParameterTo8D()
35 param->shape_[i] = data_shape[real_index--]; in PadSliceParameterTo8D()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32/ |
| D | strided_slice_fp32.c | 38 int32_t real_index = param->in_shape_length_ - 1; in PadStridedSliceParameterTo8D() local
40 if (real_index >= 0) { in PadStridedSliceParameterTo8D()
41 param->begins_[i] = begins[real_index]; in PadStridedSliceParameterTo8D()
42 param->ends_[i] = ends[real_index]; in PadStridedSliceParameterTo8D()
43 param->strides_[i] = strides[real_index]; in PadStridedSliceParameterTo8D()
44 param->in_shape_[i] = input_shape[real_index--]; in PadStridedSliceParameterTo8D()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/tbe/ |
| D | tbe_kernel_build.h | 128 std::vector<size_t> GetDeviceInputShape(const AnfNodePtr &anf_node, size_t real_index) const;
129 std::string GetDeviceInputType(const AnfNodePtr &anf_node, size_t real_index) const;
130 std::string GetDeviceInputFormat(const AnfNodePtr &anf_node, size_t real_index) const;
131 std::vector<size_t> GetDeviceOutputShape(const AnfNodePtr &anf_node, size_t real_index) const;
132 std::string GetDeviceOutputType(const AnfNodePtr &anf_node, size_t real_index) const;
133 std::string GetDeviceOutputFormat(const AnfNodePtr &anf_node, size_t real_index) const;
|
| D | tbe_kernel_build.cc | 133 std::vector<int64_t> GetOutputShapeForTbeBuild(const AnfNodePtr &anf_node, size_t real_index) { in GetOutputShapeForTbeBuild() argument
136 auto output_shape = AnfAlgo::GetOutputDetailShape(anf_node, real_index); in GetOutputShapeForTbeBuild()
150 const size_t real_index) { in GetOutputDeviceShapeForTbeBuild() argument
154 shape = GetOutputShapeForTbeBuild(anf_node, real_index); in GetOutputDeviceShapeForTbeBuild()
156 auto format = AnfAlgo::GetOutputFormat(anf_node, real_index); in GetOutputDeviceShapeForTbeBuild()
157 shape = AnfAlgo::GetOutputDeviceShapeForTbeBuild(anf_node, real_index, format); in GetOutputDeviceShapeForTbeBuild()
165 std::vector<int64_t> GetInputShapeForTbeBuild(const AnfNodePtr &anf_node, size_t real_index) { in GetInputShapeForTbeBuild() argument
167 session::KernelWithIndex kernel_with_index = AnfAlgo::GetPrevNodeOutput(anf_node, real_index); in GetInputShapeForTbeBuild()
172 const size_t real_index) { in GetInputDeviceShapeForTbeBuild() argument
175 session::KernelWithIndex kernel_with_index = AnfAlgo::GetPrevNodeOutput(anf_node, real_index); in GetInputDeviceShapeForTbeBuild()
[all …]
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/akg/ |
| D | akg_kernel_json_generator.h | 140 TypeId GetInputDataType(const AnfNodePtr &anf_node, size_t real_index) const;
141 std::vector<size_t> GetInputShape(const AnfNodePtr &anf_node, size_t real_index) const;
142 std::string GetInputFormat(const AnfNodePtr &anf_node, size_t real_index) const;
|
| D | akg_kernel_json_generator.cc | 143 TypeId AkgKernelJsonGenerator::GetInputDataType(const AnfNodePtr &anf_node, size_t real_index) cons… in GetInputDataType()
144 …ump_option_.is_before_select_kernel ? AnfAlgo::GetPrevNodeOutputInferDataType(anf_node, real_index) in GetInputDataType()
145 … : AnfAlgo::GetInputDeviceDataType(anf_node, real_index); in GetInputDataType()
148 …ize_t> AkgKernelJsonGenerator::GetInputShape(const AnfNodePtr &anf_node, size_t real_index) const { in GetInputShape()
149 …n dump_option_.is_before_select_kernel ? AnfAlgo::GetPrevNodeOutputInferShape(anf_node, real_index) in GetInputShape()
150 : AnfAlgo::GetInputDeviceShape(anf_node, real_index); in GetInputShape()
153 std::string AkgKernelJsonGenerator::GetInputFormat(const AnfNodePtr &anf_node, size_t real_index) c… in GetInputFormat()
154 …ption_.is_before_select_kernel ? kOpFormat_DEFAULT : AnfAlgo::GetInputFormat(anf_node, real_index); in GetInputFormat()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/executor/tiling/ |
| D | op_tiling_adapter.cc | 80 auto real_index = AnfAlgo::GetRealInputIndex(node, i); in ConvertInputShapeAndType() local
81 auto input_node_with_idx = AnfAlgo::GetPrevNodeOutput(node, real_index); in ConvertInputShapeAndType()
99 auto input_name = GetInputName(node, real_index); in ConvertInputShapeAndType()
|