Searched refs:matmul_type (Results 1 – 2 of 2) sorted by relevance
629 llvm::FunctionType* matmul_type = llvm::FunctionType::get( in EmitCallToRuntime() local639 module->getOrInsertFunction(fn_name, matmul_type); in EmitCallToRuntime()
791 for (const string matmul_type : {"MatMul", "SparseMatMul", "BatchMatMul"}) { in TEST_F()801 if (matmul_type == "MatMul") { in TEST_F()803 } else if (matmul_type == "SparseMatMul") { in TEST_F()805 } else if (matmul_type == "BatchMatMul") { in TEST_F()832 if (matmul_type == "BatchMatMul") { in TEST_F()