Searched defs:BatchToSpace (Results 1 – 5 of 5) sorted by relevance
34 BatchToSpace() : PrimitiveC(kNameBatchToSpace) {} in BatchToSpace() function
42 def BatchToSpace(nptype, block_size=2, input_shape=(4, 1, 2, 2)): function
21 __global__ void BatchToSpace(const size_t size, const T *input, const size_t in, in BatchToSpace() function
196 class BatchToSpace(nn.Cell): class
5026 class BatchToSpace(PrimitiveWithInfer): class