Home
last modified time | relevance | path

Searched refs:block_num_elems (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dbatchtospace_op.cc52 int64 block_num_elems = 1; in BatchToSpace() local
54 block_num_elems *= block_shape[i]; in BatchToSpace()
56 OP_REQUIRES(ctx, block_num_elems > 0, in BatchToSpace()
66 ctx, batch_size % block_num_elems == 0, in BatchToSpace()
69 block_num_elems, ")")); in BatchToSpace()
72 reshaped_shape[block_rank] = batch_size / block_num_elems; in BatchToSpace()
106 reshaped_permuted_shape[0] = batch_size / block_num_elems; in BatchToSpace()
Dspacetobatch_op.cc55 int64 block_num_elems = 1LL; in SpaceToBatch() local
66 block_num_elems *= block_shape[i]; in SpaceToBatch()
72 OP_REQUIRES(ctx, block_num_elems > 0, in SpaceToBatch()
137 output_shape[0] = batch_size * block_num_elems; in SpaceToBatch()