| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tfrt/ir/ |
| H A D | tfrt_fallback_common.h | 29 auto op_attr_array = op.op_attrs().getValue(); in VerifyExecuteOpCommon() local
30 for (auto op_attr : op_attr_array) { in VerifyExecuteOpCommon()
94 auto op_attr_array = op_attrs.getValue(); in PrintExecuteOpCommon() local
96 llvm::interleaveComma(op_attr_array, p, print_key_value); in PrintExecuteOpCommon()
102 mlir::MLIRContext *context, llvm::ArrayRef<mlir::Attribute> op_attr_array,
|
| H A D | tfrt_fallback_common.cc | 23 mlir::MLIRContext *context, llvm::ArrayRef<mlir::Attribute> op_attr_array, in GetExecuteOpAttrsCommon() argument
30 for (auto iter : op_attr_array) { in GetExecuteOpAttrsCommon()
103 llvm::SmallVector<mlir::Attribute, 4> op_attr_array; in ParseExecuteOpCommon() local
107 op_attr_array.push_back(builder.getArrayAttr({key, value})); in ParseExecuteOpCommon()
111 builder.getNamedAttr("op_attrs", builder.getArrayAttr(op_attr_array))); in ParseExecuteOpCommon()
|
| H A D | tfrt_fallback_async.cc | 248 SmallVector<Attribute, 4> op_attr_array; in parse() local
252 op_attr_array.push_back(builder.getArrayAttr({key, value})); in parse()
256 builder.getNamedAttr("op_attrs", builder.getArrayAttr(op_attr_array))); in parse()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/runtime_fallback/opdefs/ |
| H A D | tfrt_fallback_common.h | 29 auto op_attr_array = op.op_attrs().getValue(); in VerifyExecuteOpCommon() local
30 for (auto op_attr : op_attr_array) { in VerifyExecuteOpCommon()
94 auto op_attr_array = op_attrs.getValue(); in PrintExecuteOpCommon() local
96 llvm::interleaveComma(op_attr_array, p, print_key_value); in PrintExecuteOpCommon()
102 mlir::MLIRContext *context, llvm::ArrayRef<mlir::Attribute> op_attr_array,
|
| H A D | tfrt_fallback_common.cc | 23 mlir::MLIRContext *context, llvm::ArrayRef<mlir::Attribute> op_attr_array, in GetExecuteOpAttrsCommon() argument
30 for (auto iter : op_attr_array) { in GetExecuteOpAttrsCommon()
103 llvm::SmallVector<mlir::Attribute, 4> op_attr_array; in ParseExecuteOpCommon() local
107 op_attr_array.push_back(builder.getArrayAttr({key, value})); in ParseExecuteOpCommon()
111 builder.getNamedAttr("op_attrs", builder.getArrayAttr(op_attr_array))); in ParseExecuteOpCommon()
|
| H A D | tfrt_fallback_async.cc | 182 SmallVector<Attribute, 4> op_attr_array; in parseBatchFunctionOp() local
186 op_attr_array.push_back(builder.getArrayAttr({key, value})); in parseBatchFunctionOp()
190 builder.getNamedAttr("op_attrs", builder.getArrayAttr(op_attr_array))); in parseBatchFunctionOp()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/runtime_fallback/runtime/ |
| H A D | runtime_fallback_batch_kernels.cc | 53 AggregateAttr op_attr_array, in ExecuteFallbackOp() argument
62 for (size_t i = 0, e = op_attr_array.GetNumElements(); i != e; ++i) { in ExecuteFallbackOp()
63 auto pair = op_attr_array.GetAttributeOfType<AggregateAttr>(i); in ExecuteFallbackOp()
140 AggregateAttr op_attr_array, in BatchFunctionFallback() argument
181 op_attr_array, FormRef(&(*f)), std::move(exec_ctx_ptr)); in BatchFunctionFallback()
|
| H A D | runtime_fallback_kernels.cc | 1079 AggregateAttr op_attr_array, AggregateAttr op_func_attr_array, in RuntimeFallbackExecuteOp() argument
1103 tfrt::SetUpOpAttrs(op_attr_array, &op_attrs); in RuntimeFallbackExecuteOp()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/runtime_fallback/util/ |
| H A D | attr_util.cc | 667 Status SetUpAttrValueMap(tfrt::AggregateAttr op_attr_array, in SetUpAttrValueMap() argument
679 for (size_t i = 0, e = op_attr_array.GetNumElements(); i != e; ++i) { in SetUpAttrValueMap()
680 auto name_attr_pair = obtain_name_attr_pair(op_attr_array, i); in SetUpAttrValueMap()
|
| H A D | attr_util.h | 93 tensorflow::Status SetUpAttrValueMap(tfrt::AggregateAttr op_attr_array,
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/runtime_fallback/kernel/ |
| H A D | kernel_fallback_execute_compat.cc | 602 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()
|