You are an expert in writing code in many different languages. Your goal is to perform code completion for the following code, keeping in mind the rest of the code and the file meta data.
    Metadata details:
    filename: reduce.rs
    path of file: ./repos/burn/crates/burn-jit/src/tests
    
    the code of the file until where you have to start completion: #[burn_tensor_testgen::testgen(reduction)]
mod reduction {
    use super::*;
    use burn_jit::kernel::reduce::{argmax, argmin, mean_dim, sum, sum_dim, ReduceStrategy};
    use burn_tensor::{ops::IntTensorOps, Data, Distribution, Int, Shape, Tensor};

    #[test]
    fn reduction_sum_dim_should_work_with_multiple_invocations() {
        let tensor =
            Tensor::<TestBackend, 2>::random([6, 1024], Distribution::Default, &Default::default());
        let tensor_ref =
            Tensor::<ReferenceBackend, 2>::from_da
    
    fn reduction_argmin_shared_memory_medium() {
        let tensor =
            Tensor::<TestBackend, 2>::random([6, 1024], Distribution::Default, &Default::default());
        let tensor_ref =
            Tensor::<ReferenceBackend, 2>::from_data(tensor.to_data(), &Default::default());
        let reduce_dim = 1;

        let val = Tensor::<TestBackend, 2>::from_primitive(argmin::<TestRuntime, f32, f32, 2>(
            tensor.into_primitive(),
            reduce_dim,
            ReduceStrategy::SharedMemory,
        ));
        let val_ref = tensor_ref.argmin(reduce_dim);

        assert_eq!(val_ref.into_data().convert(), val.into_data());
    }