Home
last modified time | relevance | path

Searched refs:kMaxSpaceToBatchBlockDims (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dbatchtospace_op.cc115 OP_REQUIRES(context, internal_block_dims <= kMaxSpaceToBatchBlockDims, in BatchToSpaceOpCompute()
119 kMaxSpaceToBatchBlockDims)); in BatchToSpaceOpCompute()
Dspacetobatch_op.cc111 if (internal_block_dims > kMaxSpaceToBatchBlockDims) { in SpaceToBatchOpCompute()
115 kMaxSpaceToBatchBlockDims); in SpaceToBatchOpCompute()
Dspacetobatch_functor.h33 constexpr int kMaxSpaceToBatchBlockDims = 4; variable