Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/transforms/
Dembed_memref_prints.cc63 auto memref_type = memref.getType().cast<MemRefType>(); in EmitPrint() local
Dbufferize_pass.cc99 if (auto memref_type = inputs[0].getType().dyn_cast<MemRefType>()) { in CustomBufferizeTypeConverter() local
Dtf_framework_legalize_to_llvm.cc170 MemRefType memref_type = tf_alloc_op.getType(); in matchAndRewrite() local
241 MemRefType memref_type, in CreateMemRefDescriptor()
Dbufferize.cc52 auto memref_type = in matchAndRewrite() local
/external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/
Dhlo_legalize_to_lhlo.cc61 auto memref_type = in InsertDynamicAlloc() local
88 auto memref_type = in InsertAlloc() local
Dinput_inline_fusion_pass.cc271 auto memref_type = in miscFuseHelper() local
Dral_legalize_to_llvm.cc56 if (auto memref_type = t.dyn_cast<MemRefType>()) { in getTypeEncoding() local
/external/tensorflow/tensorflow/compiler/mlir/tfrt/jit/
Dtf_cpurt_passes.cc65 mlir::MemRefType memref_type = value.getType().dyn_cast<mlir::MemRefType>(); in IsContiguousMemref() local
323 auto memref_type = memref.getType().cast<mlir::MemRefType>(); in MemrefToVector() local
345 auto memref_type = memref.getType().cast<mlir::MemRefType>(); in MemrefToScalar() local
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dmhlo_to_lhlo_with_xla.cc1016 auto memref_type = type.dyn_cast<MemRefType>(); in EmitConstant() local