Searched refs:reshaped_padded_shape (Results 1 – 3 of 3) sorted by relevance
89 std::vector<int64> reshaped_padded_shape(input_rank + block_rank); in SpaceToBatch() local90 reshaped_padded_shape[0] = batch_size; in SpaceToBatch()98 reshaped_padded_shape[1 + i * 2] = padded_shape[1 + i] / block_shape[i]; in SpaceToBatch()99 reshaped_padded_shape[1 + i * 2 + 1] = block_shape[i]; in SpaceToBatch()102 reshaped_padded_shape.begin() + 1 + 2 * block_rank); in SpaceToBatch()104 xla::XlaOp reshaped_padded = xla::Reshape(padded, reshaped_padded_shape); in SpaceToBatch()115 std::vector<int64> permutation(reshaped_padded_shape.size()); in SpaceToBatch()
52 reshaped_padded_shape = [input_array.shape[0]]56 reshaped_padded_shape.append(reduced_size)58 reshaped_padded_shape.append(block_shape_value)59 reshaped_padded_shape.extend(input_array.shape[num_block_dims + 1:])62 reshaped_padded = padded.reshape(reshaped_padded_shape)
56 reshaped_padded_shape = [input_array.shape[0]]60 reshaped_padded_shape.append(reduced_size)62 reshaped_padded_shape.append(block_shape_value)63 reshaped_padded_shape.extend(input_array.shape[num_block_dims + 1:])66 reshaped_padded = padded.reshape(reshaped_padded_shape)