Home
last modified time | relevance | path

Searched defs:iteratorTypes (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Dialect/gml_st/transforms/
Dtransforms.cc122 LinalgOp linalgOp, ArrayRef<Attribute> iteratorTypes, in generateLoopNest()
201 SmallVector<Attribute, 4> iteratorTypes; in tileLinalgOpImpl() local
/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Dialect/gml_st/IR/
Dgml_st_ops.cc173 ArrayAttr iteratorTypes, in build()
184 ArrayAttr iteratorTypes, in build()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Ddot_op_emitter.cc318 llvm::SmallVector<llvm::StringRef, 4> iteratorTypes( in EmitLinalgMatmul() local