Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tfrt/jit/
H A 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
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/transforms/
H A Dtf_framework_legalize_to_llvm.cc117 MemRefType memref_type = tf_alloc_op.getType(); in matchAndRewrite() local
189 MemRefType memref_type, in CreateMemRefDescriptor()
H A Dembed_memref_prints.cc83 auto memref_type = memref.getType(); in EmitMemRefPrint() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tfrt/jit/transforms/
H A Dtf_jitrt_passes.cc26 auto memref_type = value.getType().dyn_cast<mlir::MemRefType>(); in IsContiguousMemref() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
H A Dmhlo_to_lhlo_with_xla.cc1074 auto memref_type = type.dyn_cast<MemRefType>(); in EmitConstant() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir/transforms/runtime/
H A Dcustom_call_encoding.cc842 auto memref_type = value.getType().cast<MemRefType>(); in Encode() local