Searched refs:rsum_shape (Results 1 – 1 of 1) sorted by relevance
1431 ArrayRef<int64_t> rsum_shape(rsum_shape_v); in convertSoftmaxOp() local1446 RankedTensorType::get(rsum_shape, int16_element_qtype); in convertSoftmaxOp()1448 RankedTensorType::get(rsum_shape, rewriter.getIntegerType(32)); in convertSoftmaxOp()1576 ArrayRef<int64_t> rsum_shape(rsum_shape_v); in convertSoftmaxOp() local1587 RankedTensorType::get(rsum_shape, output_type.getElementType()); in convertSoftmaxOp()1645 SmallVector<int64_t, 4> rsum_shape(output_type.getShape().begin(), in convertLogSoftmaxOp() local1647 rsum_shape[input_rank - 1] = 1; in convertLogSoftmaxOp()1649 ArrayRef<int64_t>(rsum_shape), output_type.getElementType()); in convertLogSoftmaxOp()