Searched defs:BatchToSpace (Results 1 – 5 of 5) sorted by relevance
24 void BatchToSpace(XlaOpKernelContext* ctx, const xla::XlaOp& input, in BatchToSpace() function
2368 StatusOr<HloInstruction*> ConvolutionVisitor::BatchToSpace( in BatchToSpace() function in xla::__anon5e1d62920111::ConvolutionVisitor
186 "BatchToSpace": { object971 "BatchToSpace": { object
1883 TEST_F(OpTest, BatchToSpace) { in TEST_F() argument
2916 func BatchToSpace(scope *Scope, input tf.Output, crops tf.Output, block_size int64) (output tf.Outp… func