Home
last modified time | relevance | path

Searched defs:memrefType (Results 1 – 9 of 9) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Dialect/mhlo/transforms/
H A Dhlo_legalize_to_memref.cc90 auto memrefType = in bufferize() local
211 auto memrefType = in bufferize() local
H A Dhlo_legalize_to_lhlo.cc64 auto memrefType = in insertDynamicAlloc() local
92 auto memrefType = in insertAlloc() local
H A Dhlo_legalize_to_arithmetic.cc51 auto memrefType = MemRefType::get({}, seedType); in matchAndRewrite() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Transforms/
H A Dbufferize_pass.cc116 if (auto memrefType = inputs[0].getType().dyn_cast<MemRefType>()) { in CustomBufferizeTypeConverter() local
H A Dgpu_fusion_rewrite.cc317 if (auto memrefType = shapedType.dyn_cast<MemRefType>()) in isRewritableType() local
H A Dbufferize.cc50 auto memrefType = MemRefType::get(resultType.getShape(), elementType); in matchAndRewrite() local
H A Dbuffer_packing.cc365 auto memrefType = in BufferPacking() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Dialect/gml_st/transforms/
H A Dbufferize_tiled_loop.cc105 if (auto memrefType = inputs[0].getType().dyn_cast<MemRefType>()) { in CustomBufferizeTypeConverter() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Dialect/gml_st/IR/
H A Dgml_st_ops.cc149 if (auto memrefType = sourceType.dyn_cast<MemRefType>()) { in inferReturnTypes() local