HomeSort by relevance Sort by last modified time
    Searched refs:BatchToSpaceOp (Results 1 - 2 of 2) sorted by null

  /external/tensorflow/tensorflow/core/kernels/
batchtospace_op.cc 219 class BatchToSpaceOp : public OpKernel {
221 explicit BatchToSpaceOp(OpKernelConstruction* context) : OpKernel(context) {
264 BatchToSpaceOp<CPUDevice, T>);
281 BatchToSpaceOp<GPUDevice, T>);
  /external/tensorflow/tensorflow/compiler/tf2xla/kernels/
batchtospace_op.cc 165 class BatchToSpaceOp : public XlaOpKernel {
167 explicit BatchToSpaceOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) {
188 BatchToSpaceOp);

Completed in 221 milliseconds