Searched refs:permuted_reshaped_padded (Results 1 – 5 of 5) sorted by relevance
123 xla::XlaOp permuted_reshaped_padded = in SpaceToBatch() local144 xla::XlaOp output = xla::Reshape(permuted_reshaped_padded, output_shape); in SpaceToBatch()
40 `permuted_reshaped_padded` of shape:49 4. Reshape `permuted_reshaped_padded` to flatten `block_shape` into the batch
63 permuted_reshaped_padded = np.transpose(reshaped_padded, (68 return permuted_reshaped_padded.reshape(output_shape)
67 permuted_reshaped_padded = np.transpose(reshaped_padded, (72 return permuted_reshaped_padded.reshape(output_shape)
14001 `permuted_reshaped_padded` of shape:14010 4. Reshape `permuted_reshaped_padded` to flatten `block_shape` into the batch