Searched defs:matmul_type (Results 1 – 3 of 3) sorted by relevance
206 const auto matmul_type = RankedTensorType::get({rows, cols}, element_type); in matchAndRewrite() local
660 auto matmul_type = in rewriteToBatchMatmul() local
865 llvm::FunctionType* matmul_type = llvm::FunctionType::get( in EmitCallToRuntime() local948 llvm::FunctionType* matmul_type = llvm::FunctionType::get( in EmitCallToBatchRuntime() local