Searched defs:SpaceToBatch (Results 1 – 3 of 3) sorted by relevance
24 void SpaceToBatch(XlaOpKernelContext* ctx, const xla::XlaOp& input, in SpaceToBatch() function
2955 TEST_F(OpTest, SpaceToBatch) { in TEST_F() argument
869 func SpaceToBatch(scope *Scope, input tf.Output, paddings tf.Output, block_size int64) (output tf.O… func