Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dspacetobatch_op.cc148 class SpaceToBatchNDOp : public XlaOpKernel { class
150 explicit SpaceToBatchNDOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) {} in SpaceToBatchNDOp() function in tensorflow::__anonc12dfccc0111::SpaceToBatchNDOp
/external/tensorflow/tensorflow/core/kernels/
Dspacetobatch_op.cc208 class SpaceToBatchNDOp : public OpKernel { class
210 explicit SpaceToBatchNDOp(OpKernelConstruction* context) in SpaceToBatchNDOp() function in tensorflow::SpaceToBatchNDOp