Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/cutlass_extensions/gemm/kernel/
H A DfpA_intB_gemm.h102 cutlass::gemm::GemmCoord problem_size; member
141 problem_size(problem_size), in problem_size() function
159 cutlass::gemm::GemmCoord problem_size; member
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Disotonic_regression_op.cc152 const int problem_size = solution->dimensions()[1]; in solve_pava_batch() local
/aosp_15_r20/external/ruy/ruy/
H A Dtrmul.cc270 const std::int64_t problem_size = rows_i64 * cols_i64 * depth_i64; in GetTentativeThreadCount() local
/aosp_15_r20/external/eigen/Eigen/src/Core/products/
H A DGeneralBlockPanelKernel.h279 Index problem_size = k*n*sizeof(LhsScalar); variable