Searched defs:DepthToSpace (Results 1 – 5 of 5) sorted by relevance
/third_party/mindspore/mindspore/core/ops/ |
D | depth_to_space.h | 36 DepthToSpace() : PrimitiveC(kNameDepthToSpace) { InitIOName({"x"}, {"y"}); } in DepthToSpace() function
|
/third_party/mindspore/tests/st/ops/cpu/ |
D | test_depthtospace_op.py | 43 def DepthToSpace(nptype, block_size=2, input_shape=(1, 12, 1, 1)): function
|
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/cuda_impl/ |
D | depthtospace_impl.cu | 21 __global__ void DepthToSpace(const size_t size, const T *input, const size_t in, in DepthToSpace() function
|
/third_party/mindspore/tests/st/ops/gpu/ |
D | test_depthtospace_op.py | 42 def DepthToSpace(nptype, block_size=2, input_shape=(1, 12, 1, 1)): function
|
/third_party/mindspore/mindspore/ops/operations/ |
D | array_ops.py | 4871 class DepthToSpace(PrimitiveWithInfer): class
|