Home
last modified time | relevance | path

Searched refs:ExpandDims (Results 1 – 25 of 81) sorted by relevance

1234

/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/nn/
Dflatten_gpu_kernel.cc26 MS_REG_GPU_KERNEL_ONE(ExpandDims, KernelAttr().AddInputAttr(kNumberTypeFloat64).AddOutputAttr(kNumb…
33 MS_REG_GPU_KERNEL_ONE(ExpandDims, KernelAttr().AddInputAttr(kNumberTypeFloat32).AddOutputAttr(kNumb…
38 MS_REG_GPU_KERNEL_ONE(ExpandDims, KernelAttr().AddInputAttr(kNumberTypeFloat16).AddOutputAttr(kNumb…
47 MS_REG_GPU_KERNEL_ONE(ExpandDims, KernelAttr().AddInputAttr(kNumberTypeInt64).AddOutputAttr(kNumber…
54 MS_REG_GPU_KERNEL_ONE(ExpandDims, KernelAttr().AddInputAttr(kNumberTypeInt32).AddOutputAttr(kNumber…
61 MS_REG_GPU_KERNEL_ONE(ExpandDims, KernelAttr().AddInputAttr(kNumberTypeInt16).AddOutputAttr(kNumber…
69 MS_REG_GPU_KERNEL_ONE(ExpandDims, KernelAttr().AddInputAttr(kNumberTypeUInt32).AddOutputAttr(kNumbe…
77 MS_REG_GPU_KERNEL_ONE(ExpandDims, KernelAttr().AddInputAttr(kNumberTypeUInt8).AddOutputAttr(kNumber…
84 MS_REG_GPU_KERNEL_ONE(ExpandDims, KernelAttr().AddInputAttr(kNumberTypeInt8).AddOutputAttr(kNumberT…
91 MS_REG_GPU_KERNEL_ONE(ExpandDims, KernelAttr().AddInputAttr(kNumberTypeBool).AddOutputAttr(kNumberT…
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/graph_kernel/expanders/
Dreshape.cc24 class ExpandDims : public OpExpander { class
26 ExpandDims() { in ExpandDims() function in mindspore::opt::expanders::ExpandDims
30 ~ExpandDims() = default;
52 … auto shape = MakeValue(ExpandDims::InferShape(input_x->shape, GetAxisList(this->attrs_["axis"]))); in Expand()
57 OP_EXPANDER_REGISTER("ExpandDims", ExpandDims);
60 return ExpandDims::InferShape(shape, axis); in ExpandDimsInferShape()
/third_party/mindspore/mindspore/core/ops/
Dexpand_dims.h30 class MS_CORE_API ExpandDims : public PrimitiveC {
33 ExpandDims() : PrimitiveC(kNameExpandDims) { InitIOName({"x", "axis"}, {"output"}); } in ExpandDims() function
35 ~ExpandDims() = default;
36 MS_DECLARE_PARENT(ExpandDims, PrimitiveC);
42 using PrimExpandDims = std::shared_ptr<ExpandDims>;
Dexpand_dims.cc57 REGISTER_PRIMITIVE_C(kNameExpandDims, ExpandDims);
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/
Dreshape_cpu_kernel.h102 MS_REG_CPU_KERNEL(ExpandDims, KernelAttr().AddInputAttr(kNumberTypeInt8).AddOutputAttr(kNumberTypeI…
104 MS_REG_CPU_KERNEL(ExpandDims, KernelAttr().AddInputAttr(kNumberTypeInt16).AddOutputAttr(kNumberType…
106 MS_REG_CPU_KERNEL(ExpandDims, KernelAttr().AddInputAttr(kNumberTypeInt32).AddOutputAttr(kNumberType…
108 MS_REG_CPU_KERNEL(ExpandDims, KernelAttr().AddInputAttr(kNumberTypeInt64).AddOutputAttr(kNumberType…
110 MS_REG_CPU_KERNEL(ExpandDims, KernelAttr().AddInputAttr(kNumberTypeBool).AddOutputAttr(kNumberTypeB…
112 MS_REG_CPU_KERNEL(ExpandDims, KernelAttr().AddInputAttr(kNumberTypeFloat32).AddOutputAttr(kNumberTy…
114 MS_REG_CPU_KERNEL(ExpandDims, KernelAttr().AddInputAttr(kNumberTypeUInt8).AddOutputAttr(kNumberType…
116 MS_REG_CPU_KERNEL(ExpandDims, KernelAttr().AddInputAttr(kNumberTypeUInt16).AddOutputAttr(kNumberTyp…
118 MS_REG_CPU_KERNEL(ExpandDims, KernelAttr().AddInputAttr(kNumberTypeUInt32).AddOutputAttr(kNumberTyp…
120 MS_REG_CPU_KERNEL(ExpandDims, KernelAttr().AddInputAttr(kNumberTypeUInt64).AddOutputAttr(kNumberTyp…
/third_party/mindspore/mindspore/_extends/graph_kernel/expanders/
Dbatchnorm.py18 from .expand_dims import ExpandDims
54 … 'Reshape', [input_mean], attrs={'shape': ExpandDims.infer_shape(input_mean.shape, [-1, -1])})
56 … 'Reshape', [input_scale], attrs={'shape': ExpandDims.infer_shape(input_scale.shape, [-1, -1])})
58 … 'Reshape', [input_offset], attrs={'shape': ExpandDims.infer_shape(input_offset.shape, [-1, -1])})
65 … 'Reshape', [var_add_sqrt], attrs={'shape': ExpandDims.infer_shape(var_add_sqrt.shape, [-1, -1])})
99 … 'Reshape', [mean_muls], attrs={'shape': ExpandDims.infer_shape(mean_muls.shape, [-1, -1])})
118 … 'Reshape', [y_sqrt_rec], attrs={'shape': ExpandDims.infer_shape(y_sqrt_rec.shape, [-1, -1])})
124 … 'Reshape', [input_scale], attrs={'shape': ExpandDims.infer_shape(input_scale.shape, [-1, -1])})
130 … 'Reshape', [input_offset], attrs={'shape': ExpandDims.infer_shape(input_offset.shape, [-1, -1])})
Dbatchnorm_grad.py18 from .expand_dims import ExpandDims
67 …'Reshape', [input_save_mean], attrs={'shape': ExpandDims.infer_shape(input_save_mean.shape, [-1, -…
69 … 'Reshape', [inv_variance], attrs={'shape': ExpandDims.infer_shape(inv_variance.shape, [-1, -1])})
71 … 'Reshape', [input_scale], attrs={'shape': ExpandDims.infer_shape(input_scale.shape, [-1, -1])})
83 … 'Reshape', [dgamma], attrs={'shape': ExpandDims.infer_shape(dgamma.shape, [-1, -1])})
85 … 'Reshape', [tmp_b], attrs={'shape': ExpandDims.infer_shape(tmp_b.shape, [-1, -1])})
Dexpand_dims.py20 class ExpandDims(Expander): class
/third_party/mindspore/mindspore/ccsrc/transform/graph_ir/op_declare/
Darray_ops_declare.cc81 INPUT_MAP(ExpandDims) = {{1, INPUT_DESC(x)}, {2, INPUT_DESC(axis)}};
82 ATTR_MAP(ExpandDims) = EMPTY_ATTR_MAP;
83 OUTPUT_MAP(ExpandDims) = {{0, OUTPUT_DESC(y)}};
84 REG_ADPT_DESC(ExpandDims, kNameExpandDims, ADPT_DESC(ExpandDims))
Darray_ops_declare.h42 DECLARE_OP_ADAPTER(ExpandDims)
43 DECLARE_OP_USE_OUTPUT(ExpandDims)
/third_party/mindspore/tests/ut/python/ops/
Dtest_array_ops_check.py33 self.op = P.ExpandDims()
62 'block': (P.ExpandDims(), {'exception': TypeError, 'error_keywords': ['ExpandDims']}),
67 'block': (P.ExpandDims(), {'exception': TypeError, 'error_keywords': ['ExpandDims']}),
Dtest_dynamic_shape.py45 self.expand_dims = P.ExpandDims()
79 self.expand_dims = P.ExpandDims()
/third_party/mindspore/tests/ut/python/parallel/
Dtest_expand_dims.py28 self.expand_dims = P.ExpandDims().shard(strategy2)
42 self.expand_dims = P.ExpandDims().shard(strategy1)
/third_party/mindspore/mindspore/ops/composite/
Darray_ops.py99 expand_dims_op = P.ExpandDims()
184 expand_op = P.ExpandDims()
/third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp32/
Dnon_max_suppression_fp32.cc106 void ExpandDims(std::vector<int> *shape, size_t size) { in ExpandDims() function
200 ExpandDims(&box_dims, kBoxTensorDims - box_dims.size()); in Run()
213 ExpandDims(&score_dims, kScoreTensorDims - score_dims.size()); in Run()
/third_party/mindspore/mindspore/lite/tools/converter/parser/tf/
Dtf_expand_dims_parser.cc29 auto prim = std::make_unique<ops::ExpandDims>(); in Parse()
/third_party/mindspore/mindspore/lite/tools/converter/parser/tflite/
Dtflite_expand_dims_parser.cc31 auto prim = std::make_unique<ops::ExpandDims>(); in Parse()
/third_party/mindspore/mindspore/lite/src/delegate/npu/op/
Dexpand_dims_npu.cc24 expand_dims_ = new (std::nothrow) hiai::op::ExpandDims(name_); in Init()
Dexpand_dims_npu.h47 hiai::op::ExpandDims *expand_dims_ = nullptr;
Dunsqueeze_npu.h45 hiai::op::ExpandDims *unsqueeze_ = nullptr;
/third_party/mindspore/tests/st/ops/gpu/
Dtest_expand_dims.py28 self.expand_dims = P.ExpandDims()
38 self.expand_dims = P.ExpandDims()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/infer/
Dexpand_dims_infer.c56 REG_INFER(ExpandDims, PrimType_ExpandDims, ExpandDimsInferShape)
/third_party/mindspore/tests/st/ops/graph_kernel/
Dtest_expand_dims.py26 self.expand_dims = P.ExpandDims()
/third_party/mindspore/mindspore/nn/layer/
Dpooling.py217 self.expand = P.ExpandDims()
383 self.expand = P.ExpandDims()
/third_party/mindspore/mindspore/ops/_grad/
Dgrad_array_ops.py190 @bprop_getters.register(P.ExpandDims)
413 dout = P.ExpandDims()(dout, -1)
415 indices = P.ExpandDims()(indices, -1)
500 dout = P.ExpandDims()(dout, -1)
502 indices = P.ExpandDims()(indices, -1)
533 expand_dims = P.ExpandDims()

1234