Searched refs:reshaped_permuted_shape (Results 1 – 2 of 2) sorted by relevance
105 std::vector<int64> reshaped_permuted_shape(input_rank); in BatchToSpace() local106 reshaped_permuted_shape[0] = batch_size / block_num_elems; in BatchToSpace()108 reshaped_permuted_shape[1 + i] = block_shape[i] * input_shape[1 + i]; in BatchToSpace()111 reshaped_permuted_shape.begin() + 1 + block_rank); in BatchToSpace()114 xla::Reshape(permuted, reshaped_permuted_shape); in BatchToSpace()126 std::vector<int64> end_indices = reshaped_permuted_shape; in BatchToSpace()139 " end: ", crop_end, " size ", reshaped_permuted_shape[1 + i])); in BatchToSpace()
1130 std::vector<int64_t> reshaped_permuted_shape(input_rank); in matchAndRewrite() local1132 reshaped_permuted_shape[0] = batch_size / block_num_elems; in matchAndRewrite()1134 reshaped_permuted_shape[1 + i] = in matchAndRewrite()1138 reshaped_permuted_shape.begin() + 1 + block_rank); in matchAndRewrite()1141 op.getLoc(), RankedTensorType::get(reshaped_permuted_shape, element_ty), in matchAndRewrite()1144 op.getLoc(), rewriter.getI64TensorAttr(reshaped_permuted_shape))); in matchAndRewrite()1157 std::vector<int64_t> slice_sizes = reshaped_permuted_shape; in matchAndRewrite()1175 << reshaped_permuted_shape[1 + i]; in matchAndRewrite()