Searched refs:sliceInput (Results 1 – 2 of 2) sorted by relevance
79 std::vector<Value> ConvertTFBatchMatMulOp<BatchMatMulOpType>::sliceInput( in sliceInput() function in mlir::TF::ConvertTFBatchMatMulOp286 sliceInput(input_lhs, bcast.x_batch_size(), loc, rewriter); in matchAndRewrite()288 sliceInput(input_rhs, bcast.y_batch_size(), loc, rewriter); in matchAndRewrite()
40 static std::vector<Value> sliceInput(Value value, int batch_size,