Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/linalg/
H A Dlinalg_ops_common.h211 #define REGISTER_LINALG_OP_CPU(OpName, OpClass, Scalar) \ argument
215 #define REGISTER_LINALG_OP_GPU(OpName, OpClass, Scalar) \ argument
220 #define REGISTER_LINALG_OP(OpName, OpClass, Scalar) \ argument
/aosp_15_r20/external/pytorch/aten/src/ATen/native/transformers/cuda/mem_eff_attention/
H A Dgemm_kernel_utils.h108 using OpClass = cutlass::arch::OpClassSimt; member
H A Dkernel_forward.h357 using OpClass = typename GemmType::OpClass; member
437 using OpClass = typename GemmType::OpClass; member
/aosp_15_r20/external/mesa3d/src/nouveau/codegen/
H A Dnv50_ir_target.h136 enum OpClass enum