Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
H A Dbatchtospace_op.cc166 class BatchToSpaceOp : public XlaOpKernel { class
168 explicit BatchToSpaceOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) { in BatchToSpaceOp() function in tensorflow::__anon764f8f3b0111::BatchToSpaceOp
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dbatchtospace_op.cc219 class BatchToSpaceOp : public OpKernel { class
221 explicit BatchToSpaceOp(OpKernelConstruction* context) : OpKernel(context) { in BatchToSpaceOp() function in tensorflow::BatchToSpaceOp