Home
last modified time | relevance | path

Searched defs:MatmulPlan (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/
H A Dcuda_blas_lt.h111 struct MatmulPlan { struct
112 MatmulDesc op_desc;
113 MatrixLayout a_desc;
114 MatrixLayout b_desc;
115 MatrixLayout c_desc;
116 MatrixLayout d_desc;
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dmatmul_utils.h146 MatmulPlan(se::cuda::BlasLt::MatmulPlan plan, complex128 alpha, double beta, in MatmulPlan() function