Searched refs:matmul_type (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | unroll_batch_matmul.cc | 159 auto matmul_type = RankedTensorType::get({rows, cols}, element_type); in createMatMulOps() local
172 auto matmul = rewriter.create<TF::MatMulOp>(loc, matmul_type, in createMatMulOps()
|
| D | einsum.cc | 334 auto matmul_type = in rewriteToBatchMatmul() local
337 op.getLoc(), matmul_type, lhs, rhs, rewriter.getBoolAttr(false), in rewriteToBatchMatmul()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | dot_op_emitter.cc | 798 llvm::FunctionType* matmul_type = llvm::FunctionType::get( in EmitCallToRuntime() local
805 module->getOrInsertFunction(fn_name, matmul_type); in EmitCallToRuntime()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | arithmetic_optimizer_test.cc | 716 for (const string matmul_type : in TEST_F()
728 if (matmul_type == "MatMul") { in TEST_F()
730 } else if (matmul_type == "SparseMatMul") { in TEST_F()
732 } else if (matmul_type == "BatchMatMul") { in TEST_F()
734 } else if (matmul_type == "BatchMatMulV2") { in TEST_F()
764 if (matmul_type == "BatchMatMul" || matmul_type == "BatchMatMulV2") { in TEST_F()
|