Searched refs:rsum_shape_v (Results 1 – 1 of 1) sorted by relevance
1428 SmallVector<int64_t, 4> rsum_shape_v(input_type.getShape().begin(), in convertSoftmaxOp() local1430 rsum_shape_v.push_back(1); in convertSoftmaxOp()1431 ArrayRef<int64_t> rsum_shape(rsum_shape_v); in convertSoftmaxOp()1573 SmallVector<int64_t, 4> rsum_shape_v(input_type.getShape().begin(), in convertSoftmaxOp() local1575 rsum_shape_v[input_rank - 1] = 1; in convertSoftmaxOp()1576 ArrayRef<int64_t> rsum_shape(rsum_shape_v); in convertSoftmaxOp()