Searched defs:BatchToSpace (Results 1 - 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
H A Dbatchtospace_op.cc23 void BatchToSpace(XlaOpKernelContext* ctx, function in namespace:tensorflow::__anon25405
158 BatchToSpace(ctx, ctx->Input(0), input_type(0), ctx->InputShape(0),
178 BatchToSpace(ctx, ctx->Input(0), input_type(0), ctx->InputShape(0),
185 REGISTER_XLA_OP(Name("BatchToSpace")

Completed in 120 milliseconds