Home
last modified time | relevance | path

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

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