Searched refs:u64_scalar (Results 1 – 1 of 1) sorted by relevance
167 RankedTensorType u64_scalar = RankedTensorType::get({}, u64); in matchAndRewrite() local170 rewriter.create<MulOp>(loc, u64_scalar, step_size, rng_op.delta()); in matchAndRewrite()