| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/nn/ |
| D | flatten_gpu_kernel.cc | 24 MS_REG_GPU_KERNEL_ONE(Reshape, KernelAttr().AddInputAttr(kNumberTypeFloat64).AddOutputAttr(kNumberT…
31 MS_REG_GPU_KERNEL_ONE(Reshape, KernelAttr().AddInputAttr(kNumberTypeFloat32).AddOutputAttr(kNumberT…
40 MS_REG_GPU_KERNEL_ONE(Reshape, KernelAttr().AddInputAttr(kNumberTypeFloat16).AddOutputAttr(kNumberT…
45 MS_REG_GPU_KERNEL_ONE(Reshape, KernelAttr().AddInputAttr(kNumberTypeInt64).AddOutputAttr(kNumberTyp…
52 MS_REG_GPU_KERNEL_ONE(Reshape, KernelAttr().AddInputAttr(kNumberTypeInt32).AddOutputAttr(kNumberTyp…
59 MS_REG_GPU_KERNEL_ONE(Reshape, KernelAttr().AddInputAttr(kNumberTypeInt16).AddOutputAttr(kNumberTyp…
67 MS_REG_GPU_KERNEL_ONE(Reshape, KernelAttr().AddInputAttr(kNumberTypeUInt32).AddOutputAttr(kNumberTy…
75 MS_REG_GPU_KERNEL_ONE(Reshape, KernelAttr().AddInputAttr(kNumberTypeUInt8).AddOutputAttr(kNumberTyp…
82 MS_REG_GPU_KERNEL_ONE(Reshape, KernelAttr().AddInputAttr(kNumberTypeInt8).AddOutputAttr(kNumberType…
89 MS_REG_GPU_KERNEL_ONE(Reshape, KernelAttr().AddInputAttr(kNumberTypeBool).AddOutputAttr(kNumberType…
|
| /third_party/mindspore/mindspore/core/ops/ |
| D | reshape.h | 32 class MS_CORE_API Reshape : public PrimitiveC {
35 Reshape() : PrimitiveC(kNameReshape) { InitIOName({"tensor", "shape"}, {"output"}); } in Reshape() function
37 ~Reshape() = default;
38 MS_DECLARE_PARENT(Reshape, PrimitiveC);
45 using PrimReshapePtr = std::shared_ptr<Reshape>;
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/ |
| D | reshape_cpu_kernel.h | 43 MS_REG_CPU_KERNEL(Reshape, KernelAttr().AddInputAttr(kNumberTypeInt8).AddOutputAttr(kNumberTypeInt8…
44 MS_REG_CPU_KERNEL(Reshape, KernelAttr().AddInputAttr(kNumberTypeInt16).AddOutputAttr(kNumberTypeInt…
46 MS_REG_CPU_KERNEL(Reshape, KernelAttr().AddInputAttr(kNumberTypeInt32).AddOutputAttr(kNumberTypeInt…
48 MS_REG_CPU_KERNEL(Reshape, KernelAttr().AddInputAttr(kNumberTypeInt64).AddOutputAttr(kNumberTypeInt…
50 MS_REG_CPU_KERNEL(Reshape, KernelAttr().AddInputAttr(kNumberTypeBool).AddOutputAttr(kNumberTypeBool…
51 MS_REG_CPU_KERNEL(Reshape, KernelAttr().AddInputAttr(kNumberTypeFloat32).AddOutputAttr(kNumberTypeF…
53 MS_REG_CPU_KERNEL(Reshape, KernelAttr().AddInputAttr(kNumberTypeUInt8).AddOutputAttr(kNumberTypeUIn…
55 MS_REG_CPU_KERNEL(Reshape, KernelAttr().AddInputAttr(kNumberTypeUInt16).AddOutputAttr(kNumberTypeUI…
57 MS_REG_CPU_KERNEL(Reshape, KernelAttr().AddInputAttr(kNumberTypeUInt32).AddOutputAttr(kNumberTypeUI…
59 MS_REG_CPU_KERNEL(Reshape, KernelAttr().AddInputAttr(kNumberTypeUInt64).AddOutputAttr(kNumberTypeUI…
|
| /third_party/mindspore/mindspore/ccsrc/transform/graph_ir/op_declare/ |
| D | array_ops_declare.cc | 55 INPUT_MAP(Reshape) = {{1, INPUT_DESC(x)}, {2, INPUT_DESC(shape)}};
56 ATTR_MAP(Reshape) = EMPTY_ATTR_MAP;
57 OUTPUT_MAP(Reshape) = {{0, OUTPUT_DESC(y)}};
58 REG_ADPT_DESC(Reshape, kNameReshape, ADPT_DESC(Reshape))
59 REG_ADPT_DESC(FlattenGrad, kNameFlattenGrad, ADPT_DESC(Reshape))
|
| D | array_ops_declare.h | 29 DECLARE_OP_ADAPTER(Reshape)
30 DECLARE_OP_USE_OUTPUT(Reshape)
|
| /third_party/mindspore/tests/ut/cpp/python_input/gtest_input/pre_activate/ |
| D | transpose_reshape_fusion_test.py | 19 Reshape = P.Reshape() variable
41 reshape = Reshape(transpose, (2, 2, 16, 16))
|
| D | reshape_transpose_fusion_test.py | 19 Reshape = P.Reshape() variable
40 reshape = Reshape(input0, (2, 2, 16, 16))
|
| /third_party/mindspore/tests/ut/python/parallel/ |
| D | test_reshape_unexpand.py | 54 self.reshape = P.Reshape()
77 self.reshape = P.Reshape()
100 self.reshape = P.Reshape()
123 self.reshape = P.Reshape()
147 self.reshape = P.Reshape()
171 self.reshape = P.Reshape()
195 self.reshape = P.Reshape()
229 self.reshape = P.Reshape()
253 self.reshape = P.Reshape()
|
| D | test_reshape.py | 70 self.reshape = P.Reshape().shard(strategy1)
206 self.reshape = P.Reshape()
209 self.reshape2 = P.Reshape()
221 self.reshape = P.Reshape()
224 self.reshape2 = P.Reshape()
226 self.reshape3 = P.Reshape()
240 self.reshape = P.Reshape()
243 self.reshape2 = P.Reshape()
245 self.reshape3 = P.Reshape()
259 self.reshape = P.Reshape()
[all …]
|
| D | test_auto_parallel_reshape.py | 55 self.reshape = P.Reshape()
78 self.reshape = P.Reshape()
103 self.reshape = P.Reshape()
129 self.reshape = P.Reshape()
158 self.reshape = P.Reshape()
184 self.reshape = P.Reshape()
230 self.reshape = P.Reshape()
280 self.reshape = P.Reshape()
308 self.reshape = P.Reshape()
331 self.reshape1 = P.Reshape()
[all …]
|
| D | test_reshape_optimized.py | 27 self.reshape1 = P.Reshape()
28 self.reshape2 = P.Reshape()
|
| D | test_batchmm.py | 36 self.reshape = P.Reshape()
37 self.reshape2 = P.Reshape()
|
| /third_party/mindspore/tests/ut/python/ops/ |
| D | test_array_ops_check.py | 53 self.op = P.Reshape()
127 'block': (P.Reshape(), {'exception': TypeError, 'error_keywords': ['Reshape']}),
132 'block': (P.Reshape(), {'exception': TypeError, 'error_keywords': ['Reshape']}),
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/arrays/ |
| D | gatherv2_gpu_kernel.h | 56 Reshape(); in Launch()
95 Reshape(); in Init()
127 void Reshape() { in Reshape() function
|
| D | gather_grad_gpu_kernel.h | 81 Reshape(); in Init()
100 void Reshape() { in Reshape() function
|
| D | gather_gpu_kernel.h | 80 Reshape(); in Init()
99 void Reshape() { in Reshape() function
|
| /third_party/mindspore/mindspore/parallel/nn/ |
| D | loss.py | 104 exp_sum = P.Reshape()(exp_sum, (F.shape(exp_sum)[0], 1))
109 label = P.Reshape()(label, (-1,))
118 input_mask = P.Reshape()(input_mask, (-1,))
|
| /third_party/mindspore/tests/st/networks/models/ |
| D | lenet.py | 32 self.reshape = P.Reshape()
33 self.reshape1 = P.Reshape()
|
| /third_party/mindspore/tests/ut/cpp/ops/ |
| D | test_ops_reshape.cc | 35 auto reshape = std::make_shared<Reshape>(); in TEST_F()
63 auto reshape = std::make_shared<Reshape>(); in TEST_F()
|
| /third_party/mindspore/tests/st/fusion/ |
| D | test_conv_bn1_fusion.py | 68 self.reshape = P.Reshape()
96 self.reshape = P.Reshape()
122 self.reshape = P.Reshape()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/math/ |
| D | cumprod_gpu_kernel.h | 83 Reshape(); in Init()
96 void Reshape() { in Reshape() function
|
| D | cumsum_gpu_kernel.h | 83 Reshape(); in Init()
96 void Reshape() { in Reshape() function
|
| /third_party/mindspore/mindspore/ops/composite/ |
| D | array_ops.py | 100 reshape_op = P.Reshape()
182 reshape_op = P.Reshape()
|
| /third_party/mindspore/tests/st/fl/albert/src/ |
| D | model.py | 150 self.reshape = P.Reshape()
189 self.reshape = P.Reshape()
275 self.range_mat = P.Reshape()
326 self.reshape = P.Reshape()
397 self.reshape = P.Reshape()
541 self.reshape = P.Reshape()
597 self.reshape = P.Reshape()
666 self.reshape = P.Reshape()
706 self.reshape = P.Reshape()
|
| /third_party/mindspore/tests/st/networks/models/bert/src/ |
| D | bert_model.py | 135 self.reshape = P.Reshape()
193 self.reshape = P.Reshape()
280 self.range_mat = P.Reshape()
333 self.reshape = P.Reshape()
439 self.reshape = P.Reshape()
643 self.reshape = P.Reshape()
779 self.reshape = P.Reshape()
818 self.reshape = P.Reshape()
|