Home
last modified time | relevance | path

Searched defs:DepthToSpace (Results 1 – 5 of 5) sorted by relevance

/third_party/mindspore/mindspore/core/ops/
Ddepth_to_space.h36 DepthToSpace() : PrimitiveC(kNameDepthToSpace) { InitIOName({"x"}, {"y"}); } in DepthToSpace() function
/third_party/mindspore/tests/st/ops/cpu/
Dtest_depthtospace_op.py43 def DepthToSpace(nptype, block_size=2, input_shape=(1, 12, 1, 1)): function
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/cuda_impl/
Ddepthtospace_impl.cu21 __global__ void DepthToSpace(const size_t size, const T *input, const size_t in, in DepthToSpace() function
/third_party/mindspore/tests/st/ops/gpu/
Dtest_depthtospace_op.py42 def DepthToSpace(nptype, block_size=2, input_shape=(1, 12, 1, 1)): function
/third_party/mindspore/mindspore/ops/operations/
Darray_ops.py4871 class DepthToSpace(PrimitiveWithInfer): class