Home
last modified time | relevance | path

Searched defs:attr_def (Results 1 – 18 of 18) sorted by relevance

/external/tensorflow/tensorflow/c/experimental/ops/gen/model/
Dattr_spec.cc22 AttrSpec AttrSpec::Create(const OpDef::AttrDef& attr_def) { in Create()
26 AttrSpec::AttrSpec(const OpDef::AttrDef& attr_def) { in AttrSpec()
Dop_spec.cc51 for (const OpDef::AttrDef& attr_def : op_def.attr()) { in OpSpec() local
/external/tensorflow/tensorflow/python/framework/
Dop_def_library.py55 def _SatisfiesTypeConstraint(dtype, attr_def, param_name): argument
66 def _SatisfiesLengthConstraint(length, attr_def, param_name, op_type_name): argument
73 def _SatisfiesAllowedStringsConstraint(value, attr_def, arg_name, op_type_name): argument
81 def _SatisfiesIntMinimumConstraint(value, attr_def, arg_name, op_type_name): argument
Dpython_api_info.cc142 for (const auto& attr_def : op_def.attr()) { in Initialize() local
230 const OpDef::AttrDef& attr_def, in InitializeAttribute()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/
Dupgrade_graph.cc54 [](const auto& attr_def) { in GenerateResourceSharedNameIfEmpty()
62 [](const auto& attr_def) { in GenerateResourceSharedNameIfEmpty()
Dexport_tf_dialect_op.cc105 for (const auto& attr_def : op_reg_data->op_def.attr()) { in GetUnregisteredAttrs() local
/external/tensorflow/tensorflow/java/src/gen/cc/
Dop_specs.cc103 for (const auto& attr_def : op_def_.attr()) { in TypeOf() local
128 std::pair<Type, Type> TypeResolver::TypesOf(const OpDef_AttrDef& attr_def, in TypesOf()
301 AttributeSpec CreateAttribute(const OpDef_AttrDef& attr_def, in CreateAttribute()
/external/tensorflow/tensorflow/core/common_runtime/eager/
Dattr_builder.cc198 for (const auto& attr_def : op_def->attr()) { in FillAttrValueMap() local
211 for (const OpDef::AttrDef& attr_def : op_def->attr()) { in ValueMatchesDefault() local
Dexecute.cc647 auto attr_def = signature.mutable_attr()->Add(); in BuildWrappedOpSignature() local
/external/tensorflow/tensorflow/core/runtime_fallback/kernel/
Dop_kernel_runner.cc60 for (const auto& attr_def : op_def.attr()) { in BuildNodeDef() local
/external/tensorflow/tensorflow/compiler/mlir/tfr/python/
Dtfr_gen.py107 def _get_type_from_proto(arg_def=None, attr_def=None): argument
126 def _get_type_info_from_proto(arg_def=None, attr_def=None): argument
1231 def _emit_default_constant_from_proto(self, attr_def): argument
/external/tensorflow/tensorflow/core/framework/
Dgraph_def_util.cc184 for (const OpDef::AttrDef& attr_def : op_def->attr()) { in StripDefaultAttributes() local
Dfunction.cc698 for (const auto& attr_def : op_def->attr()) { in AddDefaultAttrs() local
1835 gtl::ArraySlice<string> out_def, gtl::ArraySlice<string> attr_def, in Create()
1885 gtl::ArraySlice<string> out_def, gtl::ArraySlice<string> attr_def, in Create()
1896 gtl::ArraySlice<string> attr_def, in Define()
1964 gtl::ArraySlice<string> attr_def, in Define()
Dop_def_util_test.cc37 OpDef::AttrDef attr_def; in ADef() local
Dgraph_to_functiondef.cc285 OpDef::AttrDef* attr_def = fdef->mutable_signature()->add_attr(); in FillFunctionBody() local
Dnode_def_util.cc726 for (const auto& attr_def : op_def.attr()) { in AddDefaultsToNodeDef() local
/external/tensorflow/tensorflow/compiler/tf2xla/
Dxla_op_registry.cc233 for (const OpDef::AttrDef& attr_def : op_def->attr()) { in RegisterCompilationKernels() local
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dauto_mixed_precision.cc378 const OpDef::AttrDef* attr_def = FindAttr(attr_name, op_def); in AddNode() local
869 DataTypeSet AllowedDataTypes(const OpDef::AttrDef& attr_def) { in AllowedDataTypes()
885 const OpDef::AttrDef* attr_def = FindAttr(t_attr_id.attr_name, op_def); in AllowedDataTypes() local