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

  /external/tensorflow/tensorflow/core/kernels/
spacetobatch_op.cc 205 class SpaceToBatchNDOp : public OpKernel {
207 explicit SpaceToBatchNDOp(OpKernelConstruction* context)
258 SpaceToBatchNDOp<CPUDevice, T>); \
275 SpaceToBatchNDOp<GPUDevice, T>); \
  /external/tensorflow/tensorflow/compiler/tf2xla/kernels/
spacetobatch_op.cc 150 class SpaceToBatchNDOp : public XlaOpKernel {
152 explicit SpaceToBatchNDOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) {}
168 SpaceToBatchNDOp);

Completed in 96 milliseconds