Home
last modified time | relevance | path

Searched defs:src_shape (Results 1 – 25 of 36) sorted by relevance

12

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/cpu_kernel/format_transfer/
Dformat_transfer_transpose.cc49 bool ShapeArgValid(const std::vector<int64_t> &src_shape, const std::vector<int64_t> &perm_arg) { in ShapeArgValid()
79 bool IsTransposeArgValid(const uint8_t *src, const std::vector<int64_t> &src_shape, DataType src_da… in IsTransposeArgValid()
122 void TransShapeByPerm(const std::vector<int64_t> &src_shape, const std::vector<int64_t> &perm_arg, in TransShapeByPerm()
131 uint32_t Transpose(const uint8_t *src, const std::vector<int64_t> &src_shape, DataType src_data_typ… in Transpose()
192 uint32_t TransposeWithShapeCheck(const uint8_t *data, const std::vector<int64_t> &src_shape, in TransposeWithShapeCheck()
246 …t32_t FormatTransferTranspose::TransShape(Format src_format, const std::vector<int64_t> &src_shape, in TransShape()
Dformat_transfer_fractalz_3d.cc83 uint32_t TransShapeNcdhwToFzWithGroups(const std::vector<int64_t> &src_shape, DataType data_type, in TransShapeNcdhwToFzWithGroups()
96 uint32_t TransShapeDhwcnToFzWithGroups(const std::vector<int64_t> &src_shape, DataType data_type, in TransShapeDhwcnToFzWithGroups()
110 uint32_t TransShapeNdhwcToFzWithGroups(const std::vector<int64_t> &src_shape, DataType data_type, in TransShapeNdhwcToFzWithGroups()
261 …32_t FormatTransferFractalz3D::TransShape(Format src_format, const std::vector<int64_t> &src_shape, in TransShape()
Dformat_transfer_fractal_z.cc85 uint32_t TransShapeNchwToFzWithGroups(const std::vector<int64_t> &src_shape, DataType data_type, in TransShapeNchwToFzWithGroups()
98 uint32_t TransShapeHwcnToFzWithGroups(const std::vector<int64_t> &src_shape, DataType data_type, in TransShapeHwcnToFzWithGroups()
112 uint32_t TransShapeNhwcToFzWithGroups(const std::vector<int64_t> &src_shape, DataType data_type, in TransShapeNhwcToFzWithGroups()
262 …nt32_t FormatTransferFractalZ::TransShape(Format src_format, const std::vector<int64_t> &src_shape, in TransShape()
Dformat_transfer_ndc1hwc0.cc129 uint32_t TransShapeToNdc1hwc0(const std::vector<int64_t> &src_shape, const Format &src_format, in TransShapeToNdc1hwc0()
189 …nt32_t FormatTransferNdc1hwc0::TransShape(Format src_format, const std::vector<int64_t> &src_shape, in TransShape()
Dregister_format_transfer.h36 std::vector<int64_t> src_shape; member
Dformat_transfer_fractal_nz.cc71 uint32_t TransShapeToFracNz(const ShapeVector &src_shape, DataType data_type, ShapeVector &dst_shap… in TransShapeToFracNz()
332 uint32_t FormatTransferFractalNz::TransShape(Format src_format, const ShapeVector &src_shape, DataT… in TransShape()
389 uint32_t FormatTransferFractalNzND::TransShape(Format src_format, const ShapeVector &src_shape, Dat… in TransShape()
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/micro/coder/opcoders/nnacl/fp32/
Dsplice_fp32_coder.cc29 std::vector<int> src_shape = input_tensor_->shape(); in DoCode() local
Daffine_fp32_coder.cc56 std::vector<int> src_shape = input_tensors_.at(kInputIndex)->shape(); in PrepareSpliceOp() local
/third_party/mindspore/mindspore-src/source/mindspore/core/ops/ops_func_impl/
Dkv_cache_scatter_update.cc33 const auto &src_shape = src_shape_ptr->GetShapeVector(); in InferShape() local
Dscatter_add_ext.cc30 auto src_shape = input_args[kIndex3]->GetShape(); in InferShape() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/mkldnn/
Dlog_softmax_grad_cpu_kernel.cc48 auto src_shape = inputs[0]->GetDeviceShapeVector(); in Resize() local
Dlog_softmax_cpu_kernel.cc42 const auto &src_shape = inputs.at(kIndex0)->GetShapeVector(); in Resize() local
Dmkl_cpu_kernel.cc60 void MKLCpuKernelMod::GetPadding(const std::vector<int64_t> &src_shape, const PaddingInfo &padding_… in GetPadding()
95 void MKLCpuKernelMod::GetPadding(const std::vector<int64_t> &src_shape, const PaddingInfo &padding_… in GetPadding() argument
Dconv_cpu_kernel.cc51 auto src_shape = inputs[kIndex0]->GetDeviceShapeVector(); in Resize() local
Dconv3d_transpose_cpu_kernel.cc56 std::vector<int64_t> src_shape = outputs.at(kIndex0)->GetDeviceShapeVector(); in Resize() local
Dconv_grad_filter_cpu_kernel.cc67 auto src_shape = inputs[src_index_]->GetDeviceShapeVector(); in Resize() local
Dconv_grad_input_cpu_kernel.cc68 std::vector<int64_t> src_shape = outputs[0]->GetDeviceShapeVector(); in Resize() local
Dgru_cpu_kernel.cc89 auto src_shape = inputs[kIndex0]->GetShapeVector(); in Resize() local
Dlstm_cpu_kernel.cc94 auto src_shape = inputs[kIndex0]->GetShapeVector(); in Resize() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/optimizer/irpass/
Dreshape_eliminate.h54 auto src_shape = src_shape_abs->GetShapeTrack(); in operator() local
/third_party/mindspore/mindspore-src/source/tests/ut/cpp/ops/
Dtest_ops_scatter_add_ext.cc29 ShapeVector src_shape; member
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/tensor_layout/
Dlayout_utils.cc85 void InitShapeVec(const Shape &src_shape, Shape *tgt_shape) { in InitShapeVec()
211 bool ForwardMatching(const Shape &src_shape, const Shape &expected_tgt_shape, Shape *tgt_shape, in ForwardMatching()
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/infer/
Dcommon_infer.c57 void ShapeSet(int *dst_shape, size_t *dst_shape_size, const int *src_shape, size_t src_shape_size) { in ShapeSet()
65 bool Int64ShapeSet(int *dst_shape, size_t *dst_shape_size, const int64_t *src_shape, size_t src_sha… in Int64ShapeSet()
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/delegate/npu/
Dnpu_converter_utils.cc61 ge::Shape ConverterToNPUShape(const std::vector<int64_t> &src_shape, bool is_expand_4d) { in ConverterToNPUShape()
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/micro/coder/opcoders/nnacl/int8/
Daffine_int8_coder.cc76 std::vector<int> src_shape = input_tensors_.at(kInputIndex)->shape(); in PrepareSpliceOp() local

12