Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
H A Dspacetobatch_op.cc162 class SpaceToBatchNDOp : public XlaOpKernel { class
164 explicit SpaceToBatchNDOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) {} in SpaceToBatchNDOp() function in tensorflow::__anonbb3e23fb0111::SpaceToBatchNDOp
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dspacetobatch_op.cc220 class SpaceToBatchNDOp : public OpKernel { class
222 explicit SpaceToBatchNDOp(OpKernelConstruction* context) in SpaceToBatchNDOp() function in tensorflow::SpaceToBatchNDOp