Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Ddot_op_emitter.cc285 llvm::SmallVector<mlir::AffineExpr, 2> parallel_exprs; in EmitLinalgMatmul() local
288 parallel_exprs.push_back(mlir::getAffineDimExpr(i, context)); in EmitLinalgMatmul()
302 e = parallel_exprs[par_expr_num++]; in EmitLinalgMatmul()
308 parallel_exprs.size(), mlir::IteratorType::Parallel); in EmitLinalgMatmul()
315 /*outputs=*/{s_a(parallel_exprs)}, in EmitLinalgMatmul()