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.cc296 llvm::SmallVector<mlir::AffineExpr, 2> parallel_exprs; in EmitLinalgMatmul() local
299 parallel_exprs.push_back(mlir::getAffineDimExpr(i, context)); in EmitLinalgMatmul()
313 e = parallel_exprs[par_expr_num++]; in EmitLinalgMatmul()
319 parallel_exprs.size(), toString(mlir::IteratorType::Parallel)); in EmitLinalgMatmul()
327 {b_exprs, c_exprs, parallel_exprs}), in EmitLinalgMatmul()