"test/torchaudio_unittest/batch_consistency_test.py" did not exist on "0fa0759530204840b4a399911b6064d29630ec4d"
welford.hpp 405 Bytes
Newer Older
rocking's avatar
rocking committed
1
2
3
4
5
// SPDX-License-Identifier: MIT
// Copyright (c) 2018-2024, Advanced Micro Devices, Inc. All rights reserved.

#pragma once

ltqin's avatar
ltqin committed
6
7
#include "ck_tile/ops/welford/block/block_welford.hpp"
#include "ck_tile/ops/welford/block/block_welford_problem.hpp"
rocking's avatar
rocking committed
8
#include "ck_tile/ops/welford/thread/thread_welford.hpp"
9
#include "ck_tile/ops/common/generic_2d_block_shape.hpp"
rocking's avatar
rocking committed
10
#include "ck_tile/ops/common/tensor_layout.hpp"