Home
last modified time | relevance | path

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

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