| /third_party/mindspore/tests/st/ops/cpu/ |
| D | test_spacetodepth_op.py | 29 self.SpaceToDepth = P.SpaceToDepth(2)
42 return self.SpaceToDepth(self.x)
45 def SpaceToDepth(nptype): function
55 SpaceToDepth(np.float32)
61 SpaceToDepth(np.float16)
67 SpaceToDepth(np.int32)
73 SpaceToDepth(np.int64)
79 SpaceToDepth(np.int8)
85 SpaceToDepth(np.int16)
91 SpaceToDepth(np.uint8)
[all …]
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/arrays/ |
| D | spacetodepth_gpu_kernel.cc | 20 MS_REG_GPU_KERNEL_ONE(SpaceToDepth, KernelAttr().AddInputAttr(kNumberTypeFloat32).AddOutputAttr(kNu…
22 MS_REG_GPU_KERNEL_ONE(SpaceToDepth, KernelAttr().AddInputAttr(kNumberTypeFloat16).AddOutputAttr(kNu…
24 MS_REG_GPU_KERNEL_ONE(SpaceToDepth, KernelAttr().AddInputAttr(kNumberTypeInt32).AddOutputAttr(kNumb…
26 MS_REG_GPU_KERNEL_ONE(SpaceToDepth, KernelAttr().AddInputAttr(kNumberTypeInt64).AddOutputAttr(kNumb…
28 MS_REG_GPU_KERNEL_ONE(SpaceToDepth, KernelAttr().AddInputAttr(kNumberTypeInt16).AddOutputAttr(kNumb…
30 MS_REG_GPU_KERNEL_ONE(SpaceToDepth, KernelAttr().AddInputAttr(kNumberTypeInt8).AddOutputAttr(kNumbe…
32 MS_REG_GPU_KERNEL_ONE(SpaceToDepth, KernelAttr().AddInputAttr(kNumberTypeUInt8).AddOutputAttr(kNumb…
34 MS_REG_GPU_KERNEL_ONE(SpaceToDepth, KernelAttr().AddInputAttr(kNumberTypeUInt16).AddOutputAttr(kNum…
36 MS_REG_GPU_KERNEL_ONE(SpaceToDepth, KernelAttr().AddInputAttr(kNumberTypeUInt32).AddOutputAttr(kNum…
38 MS_REG_GPU_KERNEL_ONE(SpaceToDepth, KernelAttr().AddInputAttr(kNumberTypeUInt64).AddOutputAttr(kNum…
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/ |
| D | spacetodepth_cpu_kernel.h | 43 …SpaceToDepth, KernelAttr().SetAllSameAttr(true).AddInputAttr(kNumberTypeFloat32).AddOutputAttr(kNu…
47 …SpaceToDepth, KernelAttr().SetAllSameAttr(true).AddInputAttr(kNumberTypeFloat16).AddOutputAttr(kNu…
50 MS_REG_CPU_KERNEL_T(SpaceToDepth,
54 MS_REG_CPU_KERNEL_T(SpaceToDepth,
58 MS_REG_CPU_KERNEL_T(SpaceToDepth,
62 MS_REG_CPU_KERNEL_T(SpaceToDepth,
66 MS_REG_CPU_KERNEL_T(SpaceToDepth,
70 MS_REG_CPU_KERNEL_T(SpaceToDepth,
74 MS_REG_CPU_KERNEL_T(SpaceToDepth,
78 MS_REG_CPU_KERNEL_T(SpaceToDepth,
|
| /third_party/mindspore/tests/st/ops/gpu/ |
| D | test_spacetodepth_op.py | 28 self.SpaceToDepth = P.SpaceToDepth(2)
41 return self.SpaceToDepth(self.x)
44 def SpaceToDepth(nptype): function
55 std = P.SpaceToDepth(2)
71 SpaceToDepth(np.float32)
77 SpaceToDepth(np.float16)
83 SpaceToDepth(np.int32)
89 SpaceToDepth(np.int64)
95 SpaceToDepth(np.int8)
101 SpaceToDepth(np.int16)
[all …]
|
| /third_party/mindspore/mindspore/core/ops/ |
| D | space_to_depth.cc | 21 void SpaceToDepth::Init(const int64_t block_size, const Format &format) { in Init()
26 void SpaceToDepth::set_block_size(const int64_t block_size) { in set_block_size()
31 int64_t SpaceToDepth::get_block_size() const { in get_block_size()
36 void SpaceToDepth::set_format(const Format &format) { in set_format()
41 Format SpaceToDepth::get_format() const { in get_format()
45 REGISTER_PRIMITIVE_C(kNameSpaceToDepth, SpaceToDepth);
|
| D | space_to_depth.h | 32 class MS_CORE_API SpaceToDepth : public PrimitiveC {
35 SpaceToDepth() : PrimitiveC(kNameSpaceToDepth) { InitIOName({"x"}, {"y"}); } in SpaceToDepth() function
37 ~SpaceToDepth() = default;
38 MS_DECLARE_PARENT(SpaceToDepth, PrimitiveC);
56 using PrimSpaceToDepthPtr = std::shared_ptr<SpaceToDepth>;
|
| /third_party/mindspore/mindspore/ccsrc/transform/graph_ir/op_declare/ |
| D | transformation_ops_declare.cc | 51 INPUT_MAP(SpaceToDepth) = {{1, INPUT_DESC(x)}};
52 ATTR_MAP(SpaceToDepth) = {{"block_size", ATTR_DESC(block_size, AnyTraits<int64_t>())}};
53 OUTPUT_MAP(SpaceToDepth) = {{0, OUTPUT_DESC(y)}};
54 REG_ADPT_DESC(SpaceToDepth, kNameSpaceToDepth, ADPT_DESC(SpaceToDepth))
|
| D | transformation_ops_declare.h | 38 DECLARE_OP_ADAPTER(SpaceToDepth)
39 DECLARE_OP_USE_OUTPUT(SpaceToDepth)
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp32/ |
| D | space_to_depth_fp32.cc | 65 int SpaceToDepthCPUKernel::SpaceToDepth(int task_id) { in SpaceToDepth() function in mindspore::kernel::SpaceToDepthCPUKernel
90 auto ret = g_kernel->SpaceToDepth(task_id); in SpaceToDepthRun()
|
| D | space_to_depth_fp32.h | 30 int SpaceToDepth(int task_id);
|
| /third_party/mindspore/mindspore/lite/tools/converter/parser/onnx/ |
| D | onnx_space_to_depth_parser.cc | 25 auto prim = std::make_unique<ops::SpaceToDepth>(); in Parse()
|
| /third_party/mindspore/mindspore/lite/tools/converter/parser/tflite/ |
| D | tflite_space_to_depth_parser.cc | 32 auto prim = std::make_unique<ops::SpaceToDepth>(); in Parse()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/cuda_impl/ |
| D | spacetodepth_impl.cu | 22 __global__ void SpaceToDepth(const size_t size, const T *input, const size_t in, in SpaceToDepth() function
67 SpaceToDepth<<<GET_BLOCKS(size), GET_THREADS, 0, cuda_stream>>>( in CalSpaceToDepth()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/infer/ |
| D | space_to_depth_infer.c | 61 REG_INFER(SpaceToDepth, PrimType_SpaceToDepth, SpaceToDepthInferShape)
|
| /third_party/mindspore/mindspore/lite/schema/ |
| D | model_v0.fbs | 138 SpaceToDepth,
|
| D | ops.fbs | 162 SpaceToDepth,
933 table SpaceToDepth {
|
| D | ops_v0.fbs | 803 table SpaceToDepth {
|
| D | model_generated.h | 890 const mindspore::schema::SpaceToDepth *value_as_SpaceToDepth() const { in value_as_SpaceToDepth()
891 …ma::PrimitiveType_SpaceToDepth ? static_cast<const mindspore::schema::SpaceToDepth *>(value()) : n… in value_as_SpaceToDepth()
1640 template<> inline const mindspore::schema::SpaceToDepth *Primitive::value_as<mindspore::schema::Spa…
|
| D | model_v0_generated.h | 1029 template<> struct PrimitiveTypeTraits<mindspore::schema::v0::SpaceToDepth> {
2136 const mindspore::schema::v0::SpaceToDepth *value_as_SpaceToDepth() const {
2137 …PrimitiveType_SpaceToDepth ? static_cast<const mindspore::schema::v0::SpaceToDepth *>(value()) : n…
2837 …ate<> inline const mindspore::schema::v0::SpaceToDepth *Primitive::value_as<mindspore::schema::v0:…
4075 auto ptr = reinterpret_cast<const mindspore::schema::v0::SpaceToDepth *>(obj);
|
| /third_party/mindspore/mindspore/lite/src/ops/ |
| D | ops_func_declare.h | 409 FUNC_MSOP2SCHEMAOP_DECLARE(SpaceToDepth)
|
| D | ops_def.cc | 161 OP_TYPE(SpaceToDepth)
933 OP_SCHEMA_DEF(SpaceToDepth)
936 OP_SCHEMA_DEF_END(SpaceToDepth)
|
| /third_party/mindspore/mindspore/ops/operations/ |
| D | __init__.py | 34 … UnsortedSegmentProd, UnsortedSegmentSum, SpaceToDepth, DepthToSpace, SpaceToBatch,
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/opencl/cl/ |
| D | space_to_depth.cl | 22 __kernel void SpaceToDepth(__read_only image2d_t src_data, __write_only image2d_t dst_data, int4 in…
|
| /third_party/mindspore/tests/ut/python/ops/ |
| D | test_array_ops.py | 280 self.space_to_depth = P.SpaceToDepth(block_size)
|
| /third_party/mindspore/mindspore/ops/_grad/ |
| D | grad_array_ops.py | 825 @bprop_getters.register(P.SpaceToDepth)
839 op = P.SpaceToDepth(self.block_size)
|