Home
last modified time | relevance | path

Searched defs:SpaceToBatchOp (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
H A Dspacetobatch_op.cc182 class SpaceToBatchOp : public XlaOpKernel { class
184 explicit SpaceToBatchOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) { in SpaceToBatchOp() function in tensorflow::__anonbb3e23fb0111::SpaceToBatchOp
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dspacetobatch_op.cc236 class SpaceToBatchOp : public OpKernel { class
238 explicit SpaceToBatchOp(OpKernelConstruction* context) : OpKernel(context) { in SpaceToBatchOp() function in tensorflow::SpaceToBatchOp