Home
last modified time | relevance | path

Searched defs:rhs_layout (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dmatmul_utils.cc494 MatrixLayout rhs_layout = config.rhs_layout; in RunGemm() local
646 MatrixLayout rhs_layout = config.rhs_layout; in From() local
H A Dmatmul_utils.h102 MatrixLayout rhs_layout; member
/aosp_15_r20/external/pytorch/test/
H A Dtest_sparse_csr.py1902 def test_empty_inputs(lhs_layout, rhs_layout): argument
1915 def test_orthogonal_inputs(lhs_layout, rhs_layout): argument
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/tests/
H A Ddot_operation_test.cc332 Layout rhs_layout = LayoutUtil::MakeLayout( in TestImpl() local