Searched defs:BatchToSpace (Results 1 – 4 of 4) sorted by relevance
24 void BatchToSpace(XlaOpKernelContext* ctx, const xla::XlaOp& input, in BatchToSpace() function
1706 StatusOr<HloInstruction*> ConvolutionVisitor::BatchToSpace( in BatchToSpace() function in xla::__anon6418faad0111::ConvolutionVisitor
1229 TEST_F(OpTest, BatchToSpace) { in TEST_F() argument
1337 func BatchToSpace(scope *Scope, input tf.Output, crops tf.Output, block_size int64) (output tf.Outp… func