Home
last modified time | relevance | path

Searched refs:op_func_attr_array (Results 1 – 8 of 8) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tfrt/ir/
H A Dtfrt_fallback_common.h46 auto op_func_attr_array = op.op_func_attrs().getValue(); in VerifyFallbackExecuteOp() local
47 for (auto op_attr : op_func_attr_array) { in VerifyFallbackExecuteOp()
73 auto op_func_attr_array = op_func_attrs.getValue(); in PrintExecuteOpFuncAttribute() local
75 llvm::interleaveComma(op_func_attr_array, p, print_key_value); in PrintExecuteOpFuncAttribute()
H A Dtfrt_fallback_common.cc115 llvm::SmallVector<mlir::Attribute, 4> op_func_attr_array; in ParseExecuteOpCommon() local
119 op_func_attr_array.push_back(builder.getArrayAttr({key, value})); in ParseExecuteOpCommon()
123 "op_func_attrs", builder.getArrayAttr(op_func_attr_array))); in ParseExecuteOpCommon()
/aosp_15_r20/external/tensorflow/tensorflow/core/runtime_fallback/opdefs/
H A Dtfrt_fallback_common.h46 auto op_func_attr_array = op.op_func_attrs().getValue(); in VerifyFallbackExecuteOp() local
47 for (auto op_attr : op_func_attr_array) { in VerifyFallbackExecuteOp()
73 auto op_func_attr_array = op_func_attrs.getValue(); in PrintExecuteOpFuncAttribute() local
75 llvm::interleaveComma(op_func_attr_array, p, print_key_value); in PrintExecuteOpFuncAttribute()
H A Dtfrt_fallback_common.cc115 llvm::SmallVector<mlir::Attribute, 4> op_func_attr_array; in ParseExecuteOpCommon() local
119 op_func_attr_array.push_back(builder.getArrayAttr({key, value})); in ParseExecuteOpCommon()
123 "op_func_attrs", builder.getArrayAttr(op_func_attr_array))); in ParseExecuteOpCommon()
/aosp_15_r20/external/tensorflow/tensorflow/core/runtime_fallback/util/
H A Dattr_util.cc668 tfrt::AggregateAttr op_func_attr_array, in SetUpAttrValueMap() argument
694 for (size_t i = 0, e = op_func_attr_array.GetNumElements(); i != e; ++i) { in SetUpAttrValueMap()
695 auto name_attr_pair = obtain_name_attr_pair(op_func_attr_array, i); in SetUpAttrValueMap()
H A Dattr_util.h94 tfrt::AggregateAttr op_func_attr_array,
/aosp_15_r20/external/tensorflow/tensorflow/core/runtime_fallback/kernel/
H A Dkernel_fallback_execute_compat.cc602 tfrt::AggregateAttr op_attr_array, tfrt::AggregateAttr op_func_attr_array, in KernelFallbackCreateOp() argument
617 auto attr_builder = [op_attr_array, op_func_attr_array]( in KernelFallbackCreateOp()
619 return SetUpAttrValueMap(op_attr_array, op_func_attr_array, attr_value_map); in KernelFallbackCreateOp()
/aosp_15_r20/external/tensorflow/tensorflow/core/runtime_fallback/runtime/
H A Druntime_fallback_kernels.cc1079 AggregateAttr op_attr_array, AggregateAttr op_func_attr_array, in RuntimeFallbackExecuteOp() argument
1106 tfrt::SetUpOpFuncAttrs(op_func_attr_array, &op_attrs); in RuntimeFallbackExecuteOp()