Home
last modified time | relevance | path

Searched defs:op_attrs (Results 1 – 19 of 19) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/tfrt/ir/
Dtfrt_fallback_common.cc24 llvm::SmallVectorImpl<std::pair<llvm::StringRef, mlir::Attribute>> in GetExecuteOpAttrsCommon()
51 mlir::NamedAttrList op_attrs; in ParseExecuteOpCommon() local
Dtfrt_fallback_sync.cc81 SmallVectorImpl<std::pair<StringRef, Attribute>> *op_attrs) { in getOpAttrs()
Dtfrt_fallback_async.cc217 NamedAttrList op_attrs; in parse() local
334 SmallVectorImpl<std::pair<StringRef, Attribute>> *op_attrs) { in getOpAttrs()
Dtfrt_fallback_common.h82 auto op_attrs = op.op_attrs(); in PrintExecuteOpCommon() local
/external/tensorflow/tensorflow/core/runtime_fallback/opdefs/
Dtfrt_fallback_common.cc24 llvm::SmallVectorImpl<std::pair<llvm::StringRef, mlir::Attribute>> in GetExecuteOpAttrsCommon()
51 mlir::NamedAttrList op_attrs; in ParseExecuteOpCommon() local
Dtfrt_fallback_async.cc151 NamedAttrList op_attrs; in parseBatchFunctionOp() local
240 SmallVectorImpl<std::pair<StringRef, Attribute>> *op_attrs) { in getOpAttrs()
Dtfrt_fallback_sync.cc109 SmallVectorImpl<std::pair<StringRef, Attribute>> *op_attrs) { in getOpAttrs()
Dtfrt_fallback_common.h82 auto op_attrs = op.op_attrs(); in PrintExecuteOpCommon() local
/external/tensorflow/tensorflow/compiler/mlir/tfrt/transforms/
Dremove_device_attribute.cc52 SmallVector<std::pair<StringRef, Attribute>, 4> op_attrs; in runOnOperation() local
Dtf_to_tfrt.cc155 mlir::ArrayAttr op_attrs = corert_converter_.CreateOpAttrs(op->getAttrs()); in matchAndRewrite() local
239 mlir::ArrayAttr op_attrs, mlir::ArrayAttr op_func_attrs, in ConvertToFallbackExecuteOp()
307 mlir::ArrayAttr op_attrs, mlir::ArrayAttr op_func_attrs, in ConvertToCoreRTExecuteOp()
586 ArrayAttr op_attrs = corert_converter_.CreateOpAttrs(attr_array); in matchAndRewrite() local
848 ArrayAttr op_attrs = corert_converter_.CreateOpAttrs(op->getAttrs()); in matchAndRewrite() local
/external/tensorflow/tensorflow/core/runtime_fallback/runtime/
Druntime_fallback_batch_kernels.cc61 tfrt::OpAttrs op_attrs; in ExecuteFallbackOp() local
Druntime_fallback_kernels.cc1102 tfrt::OpAttrs op_attrs; in RuntimeFallbackExecuteOp() local
/external/tensorflow/tensorflow/core/common_runtime/eager/
Dexecute.cc485 const AbstractOpAttrs* op_attrs, string* name) { in BuildWrappedOpName()
663 auto op_attrs = op->GetOpAttrs(); in BuildWrappedOpSignature() local
724 const AbstractOpAttrs* op_attrs, in AddMixedTypeListAttrs()
755 Status PopulateRetMap(FunctionDef* fdef, const AbstractOpAttrs* op_attrs, in PopulateRetMap()
815 auto op_attrs = op->GetOpAttrs(); in WrapInCallOp() local
Deager_operation.cc488 void EagerOperation::AddAttrs(const AbstractOpAttrs* op_attrs) { in AddAttrs()
/external/tensorflow/tensorflow/core/tfrt/eager/cpp_tests/core_runtime/
Dop_handler_selector_test.cc221 void AddAttrs(const tensorflow::AbstractOpAttrs* op_attrs) override { in AddAttrs()
/external/tensorflow/tensorflow/core/tfrt/eager/
Dc_api_tfrt.cc264 void GetFuncAttr(const OpAttrs& op_attrs, const std::string& op_name, in GetFuncAttr()
1942 void OperationInterface::AddAttrs(const tensorflow::AbstractOpAttrs* op_attrs) { in AddAttrs()
/external/tensorflow/tensorflow/core/profiler/internal/
Dtfprof_node.h690 const AttrValue* op_attrs(const string& name) const { in op_attrs() function
/external/tensorflow/tensorflow/core/framework/
Dnode_def_util.cc660 std::unordered_map<string, const OpDef::AttrDef*> op_attrs; in ValidateNodeDef() local
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_nodes_test.cc7520 typename OpType::Attrs op_attrs; in CreateReduceOp() local