Searched defs:rsum_shape (Results 1 – 1 of 1) sorted by relevance
1431 ArrayRef<int64_t> rsum_shape(rsum_shape_v); in convertSoftmaxOp() local1576 ArrayRef<int64_t> rsum_shape(rsum_shape_v); in convertSoftmaxOp() local1645 SmallVector<int64_t, 4> rsum_shape(output_type.getShape().begin(), in convertLogSoftmaxOp() local