Home
last modified time | relevance | path

Searched refs:param_attr (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_compiler.cc1048 if (auto param_attr = func.getArgAttr(i, "lmhlo.params")) { in GetMlirAllocationInfo() local
1050 param_attr.cast<mlir::IntegerAttr>().getInt(), {}, in GetMlirAllocationInfo()
1071 if (auto param_attr = func.getArgAttr(i, "lmhlo.params")) { in GetMlirAllocationInfo() local
1072 o.alias_config.emplace(param_attr.cast<mlir::IntegerAttr>().getInt(), in GetMlirAllocationInfo()
/external/llvm/bindings/ocaml/llvm/
Dllvm.ml855 let param_attr p = unpack_attr (llvm_param_attr p) function
Dllvm.mli1577 (** [param_attr p] returns the attributes of parameter [p].
1580 val param_attr : llvalue -> Attribute.t list val