Searched defs:SpaceToDepth (Results 1 – 6 of 6) sorted by relevance
35 SpaceToDepth() : PrimitiveC(kNameSpaceToDepth) { InitIOName({"x"}, {"y"}); } in SpaceToDepth() function
65 int SpaceToDepthCPUKernel::SpaceToDepth(int task_id) { in SpaceToDepth() function in mindspore::kernel::SpaceToDepthCPUKernel
45 def SpaceToDepth(nptype): function
22 __global__ void SpaceToDepth(const size_t size, const T *input, const size_t in, in SpaceToDepth() function
44 def SpaceToDepth(nptype): function
4802 class SpaceToDepth(PrimitiveWithInfer): class