einsum_op_test.cpp File Reference

#include <gtest/gtest.h>
#include <ATen/core/tensor.h>
#include <ATen/ops/einsum_op.h>
#include <base/utils/gtest.h>

Include dependency graph for einsum_op_test.cpp:

Classes
class	container::op::EinsumOpTest< T >

Namespaces
namespace	container
namespace	container::op

Functions
	container::op::TYPED_TEST_SUITE (EinsumOpTest, base::utils::Types)
	container::op::TYPED_TEST (EinsumOpTest, Transform)
	container::op::TYPED_TEST (EinsumOpTest, Reduce)
	container::op::TYPED_TEST (EinsumOpTest, Stride)
	container::op::TYPED_TEST (EinsumOpTest, Inflate)
	container::op::TYPED_TEST (EinsumOpTest, ContractDot)
	container::op::TYPED_TEST (EinsumOpTest, ContractGemv)
	container::op::TYPED_TEST (EinsumOpTest, ContractGemm)
	container::op::TYPED_TEST (EinsumOpTest, TransformEllipsis)
	container::op::TYPED_TEST (EinsumOpTest, ReduceEllipsis)
	container::op::TYPED_TEST (EinsumOpTest, StrideEllipsis)
	container::op::TYPED_TEST (EinsumOpTest, InflateEllipsis)
	container::op::TYPED_TEST (EinsumOpTest, ContractGemmEllipsis)