Home
last modified time | relevance | path

Searched refs:attrs_to_ignore (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/
Dexport_tf_dialect_op.cc95 absl::flat_hash_set<absl::string_view>* attrs_to_ignore) { in GetUnregisteredAttrs() argument
112 attrs_to_ignore->insert(attr_name); in GetUnregisteredAttrs()
125 absl::flat_hash_set<absl::string_view> attrs_to_ignore; in GetAttributesToIgnore() local
131 attrs_to_ignore.insert( in GetAttributesToIgnore()
138 GetUnregisteredAttrs(inst, op_reg_data, &attrs_to_ignore)); in GetAttributesToIgnore()
145 attrs_to_ignore.insert(attr_name.data()); in GetAttributesToIgnore()
152 attrs_to_ignore.insert(attr_name.data()); in GetAttributesToIgnore()
156 attrs_to_ignore.insert("is_stateless"); in GetAttributesToIgnore()
159 attrs_to_ignore.insert("shape_invariant"); in GetAttributesToIgnore()
161 return attrs_to_ignore; in GetAttributesToIgnore()
[all …]
Dexport_graphdef.cc244 absl::flat_hash_set<absl::string_view> attrs_to_ignore = {kDeviceAttr}; in GetArgumentNode() local
245 TF_RETURN_IF_ERROR(ConvertAttributes(func_arg_i_attrs, attrs_to_ignore, in GetArgumentNode()
279 absl::flat_hash_set<absl::string_view> attrs_to_ignore = {kDeviceAttr}; in GetReturnNode() local
280 TF_RETURN_IF_ERROR(ConvertAttributes(func_res_i_attrs, attrs_to_ignore, in GetReturnNode()
670 absl::flat_hash_set<absl::string_view> attrs_to_ignore = { in ConvertLibFunction() local
674 TF_RETURN_IF_ERROR(ConvertAttributes(funcAttrs, attrs_to_ignore, in ConvertLibFunction()
689 absl::flat_hash_set<absl::string_view> attrs_to_ignore = { in ConvertLibFunction() local
692 TF_RETURN_IF_ERROR(ConvertAttributes(func_arg_i_attrs, attrs_to_ignore, in ConvertLibFunction()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/
Dexport_utils.h62 const absl::flat_hash_set<absl::string_view>& attrs_to_ignore,
Dexport_utils.cc343 const absl::flat_hash_set<absl::string_view>& attrs_to_ignore, in ConvertAttributes() argument
350 if (name == "name" || name == "device" || attrs_to_ignore.contains(name)) { in ConvertAttributes()