Searched refs:memref_type (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/transforms/ |
| D | embed_memref_prints.cc | 62 auto memref_type = memref.getType().cast<MemRefType>(); in EmitPrint() local
63 Type element_type = memref_type.getElementType(); in EmitPrint()
74 memref_type = MemRefType::get(memref_type.getShape(), element_type, in EmitPrint()
75 memref_type.getAffineMaps(), in EmitPrint()
76 memref_type.getMemorySpace()); in EmitPrint()
77 memref = b->create<IndexCastOp>(loc, memref, memref_type); in EmitPrint()
81 UnrankedMemRefType::get(element_type, memref_type.getMemorySpace()); in EmitPrint()
|
| D | tf_framework_legalize_to_llvm.cc | 78 MemRefType memref_type = tf_alloc_op.getType(); in matchAndRewrite() local
84 getMemRefDescriptorSizes(loc, memref_type, in matchAndRewrite()
91 getSizeInBytes(loc, memref_type.getElementType(), rewriter); in matchAndRewrite()
118 loc, rewriter, memref_type, allocated_byte_ptr, sizes); in matchAndRewrite()
146 MemRefType memref_type, in CreateMemRefDescriptor() argument
150 rewriter, loc, typeConverter->convertType(memref_type)); in CreateMemRefDescriptor()
154 loc, getElementPtrType(memref_type), allocated_byte_ptr); in CreateMemRefDescriptor()
159 if (memref_type.getRank() == 0) { in CreateMemRefDescriptor()
|
| D | bufferize.cc | 47 auto memref_type = in matchAndRewrite() local
52 Value buffer = rewriter.create<AllocOp>(loc, memref_type); in matchAndRewrite()
60 Value buffer = rewriter.create<AllocaOp>(loc, memref_type); in matchAndRewrite()
|
| D | bufferize_pass.cc | 95 if (auto memref_type = inputs[0].getType().dyn_cast<MemRefType>()) { in CustomBufferizeTypeConverter() local
96 assert(!memref_type.getAffineMaps().empty()); in CustomBufferizeTypeConverter()
|
| /external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/ir/ |
| D | tf_framework_ops.td | 83 OpBuilderDAG<(ins "MemRefType":$memref_type, "Value":$ctx),
86 $_state.types.push_back(memref_type);
88 OpBuilderDAG<(ins "MemRefType":$memref_type, "Value":$ctx,
91 build($_builder, $_state, memref_type, ctx);
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/ |
| D | hlo_legalize_to_lhlo.cc | 57 auto memref_type = in InsertDynamicAllocAndDealloc() local
74 return rewriter->create<AllocOp>(loc, memref_type, dynamic_operands); in InsertDynamicAllocAndDealloc()
84 auto memref_type = in InsertAlloc() local
88 auto alloc = rewriter->create<AllocOp>(loc, memref_type); in InsertAlloc()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | mhlo_to_lhlo_with_xla.cc | 902 auto memref_type = type.dyn_cast<MemRefType>(); in EmitConstant() local
903 TF_RET_CHECK(memref_type != nullptr); in EmitConstant()
918 TypeAttr::get(memref_type), initial_value, true); in EmitConstant()
936 builder_.create<GetGlobalMemrefOp>(loc, memref_type, constant_name); in EmitConstant()
|
| /external/llvm-project/mlir/test/Conversion/StandardToSPIRV/ |
| D | std-types-to-spirv.mlir | 260 // CHECK-LABEL: func @memref_type({{%.*}}: memref<3xi1>)
261 func @memref_type(%arg0: memref<3xi1>) {
|