Searched refs:reshaped_permuted (Results 1 – 4 of 4) sorted by relevance
41 3. Reshape `permuted` to produce `reshaped_permuted` of shape53 `reshaped_permuted` according to `crops` to produce the output of shape:
113 xla::XlaOp reshaped_permuted = in BatchToSpace() local142 xla::Slice(reshaped_permuted, start_indices, end_indices, strides); in BatchToSpace()
1140 auto reshaped_permuted = rewriter.create<TF::ReshapeOp>( in matchAndRewrite() local1180 op, RankedTensorType::get(slice_sizes, element_ty), reshaped_permuted, in matchAndRewrite()
1125 3. Reshape `permuted` to produce `reshaped_permuted` of shape1137 `reshaped_permuted` according to `crops` to produce the output of shape: