Searched refs:unranked_memref (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/transforms/ |
| D | embed_memref_prints.cc | 82 Value unranked_memref = b->create<MemRefCastOp>(loc, memref, unranked_type); in EmitPrint() local
85 emitCallToPrint(loc, "print_memref_f32", unranked_memref, b); in EmitPrint()
89 emitCallToPrint(loc, "print_memref_f64", unranked_memref, b); in EmitPrint()
93 emitCallToPrint(loc, "print_memref_i32", unranked_memref, b); in EmitPrint()
97 emitCallToPrint(loc, "print_memref_i64", unranked_memref, b); in EmitPrint()
|
| /external/llvm-project/mlir/test/Bindings/Python/ |
| D | ir_types.py | 354 unranked_memref = UnrankedMemRefType.get(f32, 2)
356 print("unranked memref type:", unranked_memref)
358 invalid_rank = unranked_memref.rank
365 invalid_is_dynamic_dim = unranked_memref.is_dynamic_dim(0)
372 invalid_get_dim_size = unranked_memref.get_dim_size(1)
|
| /external/llvm-project/mlir/test/Conversion/StandardToSPIRV/ |
| D | std-types-to-spirv.mlir | 448 // CHECK-LABEL: func @unranked_memref
450 func @unranked_memref(%arg0: memref<*xi32>) { return }
|