Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/
Ddevice_util.cc66 mlir::ArrayAttr array_attr, in GetDevicesFromOp() argument
70 for (auto& kv : llvm::enumerate(array_attr)) { in GetDevicesFromOp()
150 if (auto array_attr = devices_attr.dyn_cast<mlir::ArrayAttr>()) { in GetDevicesFromOp() local
151 return GetDevicesFromOp(op, array_attr, devices); in GetDevicesFromOp()
/external/tensorflow/tensorflow/compiler/mlir/lite/utils/
Dperception_ops_utils.cc48 ArrayAttr array_attr = attrs.get(attr_name).dyn_cast_or_null<ArrayAttr>(); in GetIntegerArraySafe() local
49 if (array_attr == nullptr || array_attr.size() != N) { in GetIntegerArraySafe()
56 for (Attribute integer_attr : array_attr.getValue()) { in GetIntegerArraySafe()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dexecutor_tpuv1_outline_tpu_island.cc159 if (auto array_attr = attr.second.dyn_cast<ArrayAttr>()) { in runOnOperation() local
160 for (const Attribute &attribute : array_attr) { in runOnOperation()
/external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/transforms/
Dtf_framework_legalize_to_llvm.cc194 auto &array_attr = attr.getValue(); in ConvertI32ArrayAttrToStackAllocatedArray() local
196 loc, llvm_i32_type, rewriter->getI32IntegerAttr(array_attr.size())); in ConvertI32ArrayAttrToStackAllocatedArray()
200 for (auto &dim : llvm::enumerate(array_attr)) { in ConvertI32ArrayAttrToStackAllocatedArray()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/c/
Dc_api_unified_experimental_mlir.cc408 ArrayAttr array_attr = attr.dyn_cast<ArrayAttr>(); in Create() local
409 if (!array_attr) in Create()
413 for (Attribute attr : array_attr) { in Create()
/external/tensorflow/tensorflow/compiler/mlir/tfr/ir/
Dtfr_ops.cc508 auto array_attr = rewriter.getArrayAttr(array_list); in matchAndRewrite() local
509 rewriter.replaceOpWithNewOp<TFR::ConstOp>(bl_op, array_attr); in matchAndRewrite()
/external/clang/test/Parser/
Dcxx0x-attributes.cpp47 int array_attr [1] [[]]; variable
/external/llvm-project/clang/test/Parser/
Dcxx0x-attributes.cpp47 int array_attr [1] [[]]; variable
/external/llvm-project/mlir/test/mlir-tblgen/
Dop-attribute.td111 ArrayAttr:$array_attr,
149 // DEF: ::mlir::ArrayAttr BOp::array_attr()
/external/tensorflow/tensorflow/compiler/mlir/lite/
Dflatbuffer_export.cc1563 auto array_attr = attr.second.dyn_cast_or_null<mlir::ArrayAttr>(); in GetStringsFromDictionaryAttr() local
1564 if (!array_attr || array_attr.empty()) continue; in GetStringsFromDictionaryAttr()
1565 auto string_attr = array_attr[0].dyn_cast_or_null<mlir::StringAttr>(); in GetStringsFromDictionaryAttr()
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dlegalize_tf.cc130 auto array_attr = attr.cast<ArrayAttr>(); in GetI64ArrayAttrValues() local
131 values->reserve(array_attr.getValue().size()); in GetI64ArrayAttrValues()
132 for (Attribute val : array_attr.getValue()) in GetI64ArrayAttrValues()