Home
last modified time | relevance | path

Searched refs:createMatMulOps (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dunroll_batch_matmul.h46 static TF::PackOp createMatMulOps(const std::vector<Value>& sliced_lhs,
Dunroll_batch_matmul.cc155 TF::PackOp ConvertTFBatchMatMulOp<BatchMatMulOpType>::createMatMulOps( in createMatMulOps() function in mlir::TF::ConvertTFBatchMatMulOp
293 createMatMulOps(sliced_lhs, sliced_rhs, bcast, lhs_shape[dims_a - 2], in matchAndRewrite()