Searched refs:reshaped_padded (Results 1 – 5 of 5) sorted by relevance
104 xla::XlaOp reshaped_padded = xla::Reshape(padded, reshaped_padded_shape); in SpaceToBatch() local124 xla::Transpose(reshaped_padded, permutation); in SpaceToBatch()
29 2. Reshape `padded` to `reshaped_padded` of shape:39 3. Permute dimensions of `reshaped_padded` to produce
62 reshaped_padded = padded.reshape(reshaped_padded_shape)63 permuted_reshaped_padded = np.transpose(reshaped_padded, (
66 reshaped_padded = padded.reshape(reshaped_padded_shape)67 permuted_reshaped_padded = np.transpose(reshaped_padded, (
13990 2. Reshape `padded` to `reshaped_padded` of shape:14000 3. Permute dimensions of `reshaped_padded` to produce