Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/linalg/
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
/external/pytorch/aten/src/ATen/native/transformers/cuda/mem_eff_attention/
Dgemm_kernel_utils.h108 using OpClass = cutlass::arch::OpClassSimt; member
Dkernel_forward.h357 using OpClass = typename GemmType::OpClass; member
437 using OpClass = typename GemmType::OpClass; member
/external/mesa3d/src/nouveau/codegen/
Dnv50_ir_target.h136 enum OpClass enum