pub fn mask_fill<R: CubeRuntime>(
input: CubeTensor<R>,
mask: CubeTensor<R>,
value: InputScalar,
strategy: MaskFillStrategy,
dtype_bool: DType,
) -> CubeTensor<R>Expand description
Execute the mask fill kernel with the given strategy.