Searched refs:block_num_elems (Results 1 – 5 of 5) sorted by relevance
52 int64 block_num_elems = 1; in BatchToSpace() local54 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()
55 int64 block_num_elems = 1LL; in SpaceToBatch() local66 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()
1061 int64_t block_num_elems = 1; in matchAndRewrite() local1064 block_num_elems *= val.getSExtValue(); in matchAndRewrite()1067 if (block_num_elems <= 0) { in matchAndRewrite()1083 reshaped_shape[block_rank] = batch_size / block_num_elems; in matchAndRewrite()1132 reshaped_permuted_shape[0] = batch_size / block_num_elems; in matchAndRewrite()
830 int block_num_elems = 1; in convertSpaceToBatchNDOp() local916 block_num_elems *= block_shape_val; in convertSpaceToBatchNDOp()974 a4_reshape_shape[0] = batch_size * block_num_elems; in convertSpaceToBatchNDOp()1100 int block_num_elems = 1; in convertBatchToSpaceNDOp() local1107 block_num_elems *= block_shape_val; in convertBatchToSpaceNDOp()1137 a1_shape[block_rank] = input_shape[0] / block_num_elems; in convertBatchToSpaceNDOp()1190 a4_shape[0] = input_shape[0] / block_num_elems; in convertBatchToSpaceNDOp()
754 size_t block_num_elems = 1759 block_num_elems *= %block.as_constant()[i]