Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dbatchtospace_op.cc146 class BatchToSpaceNDOp : public XlaOpKernel { class
148 explicit BatchToSpaceNDOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) {} in BatchToSpaceNDOp() function in tensorflow::__anon7c3f680c0111::BatchToSpaceNDOp
/external/tensorflow/tensorflow/core/kernels/
Dbatchtospace_op.cc204 class BatchToSpaceNDOp : public OpKernel { class
206 explicit BatchToSpaceNDOp(OpKernelConstruction* context) in BatchToSpaceNDOp() function in tensorflow::BatchToSpaceNDOp