Home
last modified time | relevance | path

Searched refs:array_attr (Results 1 – 16 of 16) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/lite/utils/
Dperception_ops_utils.cc48 ArrayAttr array_attr = attrs.get(attr_name).dyn_cast_or_null<ArrayAttr>(); in HasIntegerArrayWithSize() local
49 if (array_attr == nullptr || array_attr.size() != N) { in HasIntegerArrayWithSize()
54 for (Attribute integer_attr : array_attr.getValue()) { in HasIntegerArrayWithSize()
69 ArrayAttr array_attr = attrs.get(attr_name).dyn_cast_or_null<ArrayAttr>(); in GetIntegerArraySafe() local
70 if (array_attr == nullptr || array_attr.size() != N) { in GetIntegerArraySafe()
77 for (Attribute integer_attr : array_attr.getValue()) { in GetIntegerArraySafe()
/external/tensorflow/tensorflow/compiler/mlir/tfrt/analysis/
Dcompatibility_analysis.cc124 if (auto array_attr = attr.second.dyn_cast<mlir::ArrayAttr>()) { in AnalyzeOpAttributes() local
125 if (!array_attr.empty()) { in AnalyzeOpAttributes()
126 if (array_attr[0].isa<mlir::ElementsAttr>()) return true; in AnalyzeOpAttributes()
128 if (array_attr[0].isa<mlir::StringAttr>()) return true; in AnalyzeOpAttributes()
130 if (array_attr[0].isa<mlir::FlatSymbolRefAttr>()) return true; in AnalyzeOpAttributes()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/
Ddevice_util.cc63 mlir::ArrayAttr array_attr, in GetDevicesFromOp() argument
67 for (auto& kv : llvm::enumerate(array_attr)) { in GetDevicesFromOp()
147 if (auto array_attr = devices_attr.dyn_cast<mlir::ArrayAttr>()) { in GetDevicesFromOp() local
148 return GetDevicesFromOp(op, array_attr, devices); in GetDevicesFromOp()
/external/tensorflow/tensorflow/core/runtime_fallback/util/
Dattr_util.cc619 Status SetUpListAttr(tfrt::ArrayAttr array_attr, in SetUpListAttr() argument
624 if (array_attr.GetNumElements() == 0) { in SetUpListAttr()
628 tfrt::BEFAttributeType element_type = array_attr.GetElementType(); in SetUpListAttr()
633 for (auto value : array_attr.GetValue<bool>()) { in SetUpListAttr()
639 for (auto value : array_attr.GetValue<int64_t>()) { in SetUpListAttr()
645 for (auto value : array_attr.GetValue<float>()) { in SetUpListAttr()
656 for (auto value : array_attr.GetValue<tfrt::DType>()) { in SetUpListAttr()
687 } else if (auto array_attr = attr_value.dyn_cast<tfrt::ArrayAttr>()) { in SetUpAttrValueMap() local
688 TF_RETURN_IF_ERROR(SetUpListAttr(array_attr, &tf_attr)); in SetUpAttrValueMap()
/external/tensorflow/tensorflow/core/runtime_fallback/runtime/
Druntime_fallback_batch_kernels.cc74 auto array_attr = attr.cast<tfrt::ArrayAttr>(); in ExecuteFallbackOp() local
75 op_attrs.SetRaw(key, array_attr.GetElements(), type, in ExecuteFallbackOp()
76 array_attr.GetNumElements(), in ExecuteFallbackOp()
/external/ltp/lib/
Dtst_fd.c206 union bpf_attr array_attr = { in open_bpf_map() local
213 fd->fd = syscall(__NR_bpf, BPF_MAP_CREATE, &array_attr, sizeof(array_attr)); in open_bpf_map()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dexecutor_tpuv1_outline_tpu_island.cc184 if (auto array_attr = attr.getValue().dyn_cast<ArrayAttr>()) { in runOnOperation() local
185 for (const Attribute &attribute : array_attr) { in runOnOperation()
Dtfg-to-tfe.cc58 mlir::ArrayAttr array_attr, llvm::SmallVector<mlir::Type> &output_types, in FilterTfgSpecificArgResultAttributes() argument
61 types, array_attr.template getAsRange<mlir::DictionaryAttr>())) { in FilterTfgSpecificArgResultAttributes()
/external/tensorflow/tensorflow/compiler/mlir/tfrt/transforms/
Dcorert_converter.cc325 if (auto array_attr = attr.dyn_cast<mlir::ArrayAttr>()) { in ConvertAttribute() local
327 attrs.reserve(array_attr.size()); in ConvertAttribute()
328 for (auto attr : array_attr) { in ConvertAttribute()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/c/
Dc_api_unified_experimental_mlir.cc409 ArrayAttr array_attr = attr.dyn_cast<ArrayAttr>(); in Create() local
410 if (!array_attr) in Create()
414 for (Attribute attr : array_attr) { in Create()
/external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/transforms/
Dtf_framework_legalize_to_llvm.cc76 auto &array_attr = attr.getValue(); in ConvertArrayAttrToStackAllocatedArray() local
78 loc, size_ty, rewriter->getIntegerAttr(size_ty, array_attr.size())); in ConvertArrayAttrToStackAllocatedArray()
81 for (auto &e : llvm::enumerate(array_attr)) { in ConvertArrayAttrToStackAllocatedArray()
/external/clang/test/Parser/
Dcxx0x-attributes.cpp47 int array_attr [1] [[]]; variable
/external/tensorflow/tensorflow/core/ir/
Dtf_op_names.cc422 auto array_attr = op->getAttrOfType<ArrayAttr>("T"); in IsIdentityNSingleInput() local
423 if (!array_attr) return false; in IsIdentityNSingleInput()
425 return array_attr.size() == 1; in IsIdentityNSingleInput()
/external/tensorflow/tensorflow/compiler/mlir/tfr/ir/
Dtfr_ops.cc572 auto array_attr = rewriter.getArrayAttr(array_list); in matchAndRewrite() local
573 rewriter.replaceOpWithNewOp<TFR::ConstOp>(bl_op, array_attr); in matchAndRewrite()
/external/tensorflow/tensorflow/compiler/mlir/lite/
Dflatbuffer_export.cc1728 auto array_attr = attr.getValue().dyn_cast_or_null<mlir::ArrayAttr>(); in GetStringsFromDictionaryAttr() local
1729 if (!array_attr || array_attr.empty()) continue; in GetStringsFromDictionaryAttr()
1730 auto string_attr = array_attr[0].dyn_cast_or_null<mlir::StringAttr>(); in GetStringsFromDictionaryAttr()
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dlegalize_tf.cc84 auto array_attr = attr.cast<ArrayAttr>(); in GetI64ArrayAttrValues() local
85 values->reserve(array_attr.getValue().size()); in GetI64ArrayAttrValues()
86 for (Attribute val : array_attr.getValue()) in GetI64ArrayAttrValues()