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

  /external/tensorflow/tensorflow/core/kernels/
batchtospace_op.cc 113 const int internal_block_dims = local
115 OP_REQUIRES(context, internal_block_dims <= kMaxSpaceToBatchBlockDims,
118 internal_block_dims, " but must not exceed ",
121 if (internal_block_dims == 0) {
127 // having this shape, of rank 2 + internal_block_dims.
131 // having this shape, of rank 2 + internal_block_dims.
184 switch (internal_block_dims) {
spacetobatch_op.cc 109 const int internal_block_dims = local
111 OP_REQUIRES(context, internal_block_dims <= kMaxSpaceToBatchBlockDims,
114 internal_block_dims, " but must not exceed ",
117 if (internal_block_dims == 0) {
123 // having this shape, of rank 2 + internal_block_dims.
127 // having this shape, of rank 2 + internal_block_dims.
183 switch (internal_block_dims) {

Completed in 455 milliseconds