Home
last modified time | relevance | path

Searched refs:type_attr (Results 1 – 25 of 34) sorted by relevance

12

/external/tensorflow/tensorflow/core/ops/compat/
Dops_history.v0.pbtxt32 type_attr: "T"
36 type_attr: "T"
55 type_attr: "T"
59 type_attr: "T"
88 type_attr: "dtype"
150 type_attr: "dtype"
179 type_attr: "T"
183 type_attr: "T"
205 type_attr: "T"
209 type_attr: "T"
[all …]
Dops_history.v1.pbtxt22 type_attr: "T"
26 type_attr: "T"
46 type_attr: "T"
50 type_attr: "T"
71 type_attr: "T"
75 type_attr: "T"
96 type_attr: "T"
101 type_attr: "T"
144 type_attr: "T"
149 type_attr: "T"
[all …]
Dops_history.v2.pbtxt22 type_attr: "T"
26 type_attr: "T"
46 type_attr: "T"
50 type_attr: "T"
71 type_attr: "T"
75 type_attr: "T"
96 type_attr: "T"
101 type_attr: "T"
144 type_attr: "T"
149 type_attr: "T"
[all …]
/external/tensorflow/tensorflow/core/ops/
Dops.pbtxt22 type_attr: "T"
26 type_attr: "T"
47 type_attr: "T"
52 type_attr: "T"
105 type_attr: "dtype"
170 type_attr: "dtype"
202 type_attr: "T"
206 type_attr: "T"
229 type_attr: "T"
233 type_attr: "T"
[all …]
/external/tensorflow/tensorflow/python/framework/
Dop_def_library.py74 if arg.type_attr: num += 1
243 if arg.type_attr:
244 attr_type = _Attr(op_def, arg.type_attr).type
248 (arg.type_attr, op_def.name, attr_type))
254 (arg.type_attr, op_def.name, attr_type))
450 if input_arg.type_attr in attrs:
451 dtype = attrs[input_arg.type_attr]
460 if dtype is None and input_arg.type_attr in default_type_attr_map:
461 default_dtype = default_type_attr_map[input_arg.type_attr]
494 elif input_arg.type_attr in attrs:
[all …]
Dpython_op_gen_internal.cc220 original_arg = gtl::FindOrNull(inferred_attrs, arg.type_attr()); in ArgTypeName()
226 arg.type_attr(), "`."); in ArgTypeName()
228 const OpDef::AttrDef* attr = FindAttr(arg.type_attr(), op_def); in ArgTypeName()
243 } else if (!arg.type_attr().empty() || !arg.type_list_attr().empty()) { in ArgTypeName()
245 const string attr_name = is_list ? arg.type_list_attr() : arg.type_attr(); in ArgTypeName()
517 if (!arg.type_attr().empty()) { in Code()
518 gtl::InsertIfNotPresent(&inferred_attrs_, arg.type_attr(), arg.name()); in Code()
Dfunction_def_to_graph.py201 elif arg_def.type_attr or arg_def.type != types_pb2.DT_INVALID:
/external/tensorflow/tensorflow/go/genop/internal/
Dgenop_test.go88 type_attr: "T"
92 type_attr: "T"
96 type_attr: "T"
141 type_attr: "SrcT"
145 type_attr: "DstT"
314 type_attr: "T"
318 type_attr: "T"
322 type_attr: "T"
358 type_attr: "T"
363 type_attr: "out_type"
/external/tensorflow/tensorflow/python/framework/testdata/
Dmetrics_export_meta_graph.pb7 type_attr: "T"
12 type_attr: "T"
16 type_attr: "T"
43 type_attr: "T"
48 type_attr: "T"
52 type_attr: "T"
89 type_attr: "SrcT"
93 type_attr: "DstT"
108 type_attr: "dtype"
167 type_attr: "T"
[all …]
/external/tensorflow/tensorflow/cc/saved_model/testdata/half_plus_two_pbtxt/00000123/
Dsaved_model.pbtxt9 type_attr: "T"
13 type_attr: "T"
17 type_attr: "T"
43 type_attr: "T"
48 type_attr: "T"
52 type_attr: "T"
79 type_attr: "dtype"
94 type_attr: "T"
98 type_attr: "T"
127 type_attr: "T"
[all …]
/external/tensorflow/tensorflow/core/framework/
Dop_def_util.cc41 return arg.type() != DT_INVALID || !arg.type_attr().empty() || in HasAttrStyleType()
215 (!arg.type_attr().empty() ? 1 : 0) == in ValidateArg()
220 (!arg.type_attr().empty() ? 1 : 0) + in ValidateArg()
227 if (!arg.type_attr().empty()) { in ValidateArg()
228 const OpDef::AttrDef* attr = FindAttr(arg.type_attr(), op_def); in ValidateArg()
229 VALIDATE(attr != nullptr, "No attr with name '", arg.type_attr(), "'", in ValidateArg()
380 strings::StrAppend(&ret, arg.type_attr()); in SummarizeArgs()
578 gtl::FindPtrOrNull(old_attrs, arg.type_attr()); in ComputeArgSignature()
581 strings::StrAppend(&type, arg.type_attr()); in ComputeArgSignature()
586 gtl::FindPtrOrNull(new_attrs, arg.type_attr()); in ComputeArgSignature()
Dfake_input.cc147 } else if (!arg_->type_attr().empty()) { in GetDataType()
148 Status status = GetNodeAttr(*node_def_, arg_->type_attr(), dt); in GetDataType()
151 const OpDef::AttrDef* attr = FindAttr(arg_->type_attr(), *op_def_); in GetDataType()
Dop_def.proto31 // "type_attr" field is set to the name of an attr with type "type".
34 // either the "type" or "type_attr" field will be set as for
39 string type_attr = 4; // if specified, attr must have type "type" field
42 // type, type_attr, and number_attr may be specified.
Dnode_def_util.cc359 if (!arg_def.type_attr().empty()) { in AddArgToSig()
361 TF_RETURN_IF_ERROR(GetNodeAttr(node_def, arg_def.type_attr(), &dtype)); in AddArgToSig()
373 } else if (!arg_def.type_attr().empty()) { in AddArgToSig()
376 AttrSlice(node_def).Find(arg_def.type_attr(), &attr_value)); in AddArgToSig()
568 } else if (!arg_def.type_attr().empty() || arg_def.type() != DT_INVALID) { in ComputeArgRange()
Dnode_def_builder.cc132 Attr(input_arg->type_attr(), BaseType(dt)); in SingleInput()
151 Attr(input_arg->type_attr(), base); in ListInput()
/external/tensorflow/tensorflow/core/grappler/utils/
Dfunctions.cc281 const AttrValue* type_attr = func_instantiation_attr_.Find(type_attr_name); in GetTypeAttr() local
282 if (type_attr == nullptr) { in GetTypeAttr()
285 } else if (type_attr->type() == DT_INVALID) { in GetTypeAttr()
289 *data_type = type_attr->type(); in GetTypeAttr()
305 TF_RETURN_IF_ERROR(GetTypeAttr(arg.type_attr(), data_type)); in GetArgType()
398 return !arg.type_attr().empty() || !arg.number_attr().empty() || in HasParametrizedType()
435 type_parameters->find(arg.type_attr()) == type_parameters->end()) { in InstantiationTypeParameters()
438 type_parameters->insert({arg.type_attr(), data_type}); in InstantiationTypeParameters()
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dremapper.cc139 const string& type_attr = "T") { in HaveSameDataType() argument
140 DataType lhs_attr = GetDataTypeFromAttr(*lhs, type_attr); in HaveSameDataType()
141 DataType rhs_attr = GetDataTypeFromAttr(*rhs, type_attr); in HaveSameDataType()
148 const string& type_attr = "T") { in HasDataType() argument
149 DataType dtype = GetDataTypeFromAttr(*node, type_attr); in HasDataType()
154 const string& type_attr = "T") { in IsCpuCompatibleDataType() argument
155 DataType dtype = GetDataTypeFromAttr(*node, type_attr); in IsCpuCompatibleDataType()
160 const string& type_attr = "T") { in IsGpuCompatibleDataType() argument
161 DataType dtype = GetDataTypeFromAttr(*node, type_attr); in IsGpuCompatibleDataType()
Dfunction_api_info.cc95 if (arg1.type_attr() != arg2.type_attr()) return false; in IsSameArgDef()
/external/tensorflow/tensorflow/c/eager/
Dc_api.cc250 const std::string& type_attr = input_def.type_attr(); in OpInferSingleInputAttrs() local
251 if (!type_attr.empty() && ictx->attrs.find(type_attr) == ictx->attrs.end()) { in OpInferSingleInputAttrs()
252 op->operation.MutableAttrs()->Set(type_attr, input->handle->dtype); in OpInferSingleInputAttrs()
253 ictx->attrs.insert(type_attr); in OpInferSingleInputAttrs()
267 if (ictx->attrs.find(input_def.type_attr()) == ictx->attrs.end()) { in OpInferSingleTypeInputListAttrs()
268 op->operation.MutableAttrs()->Set(input_def.type_attr(), in OpInferSingleTypeInputListAttrs()
270 ictx->attrs.insert(input_def.type_attr()); in OpInferSingleTypeInputListAttrs()
298 } else if (!input_def.type_attr().empty() && in OpInferInputListAttrs()
/external/tensorflow/tensorflow/java/src/gen/cc/
Dop_specs.cc98 } else if (!arg_def.type_attr().empty()) { in TypeOf()
100 if (IsAttributeVisited(arg_def.type_attr())) { in TypeOf()
101 type = visited_attrs_.at(arg_def.type_attr()); in TypeOf()
104 if (attr_def.name() == arg_def.type_attr()) { in TypeOf()
/external/tensorflow/tensorflow/compiler/tf2xla/
Dxla_op_registry.cc262 for (const string& type_attr : type_attrs) { in RegisterCompilationKernels() local
264 attr_constraint->set_name(type_attr); in RegisterCompilationKernels()
268 const OpDef::AttrDef& op_def_attr = *FindAttr(type_attr, *op_def); in RegisterCompilationKernels()
274 op_registration->type_constraints.find(type_attr); in RegisterCompilationKernels()
/external/tensorflow/tensorflow/js/ops/
Dts_op_gen.cc123 if (!op_def_arg->type_attr().empty()) { in ProcessArgs()
124 AddAttrForArg(op_def_arg->type_attr(), i); in ProcessArgs()
/external/tensorflow/tensorflow/core/grappler/
Dutils.cc312 DataType GetDataTypeFromAttr(const NodeDef& node, const string& type_attr) { in GetDataTypeFromAttr() argument
313 if (!node.attr().count(type_attr)) { in GetDataTypeFromAttr()
316 const auto& attr = node.attr().at(type_attr); in GetDataTypeFromAttr()
Dutils.h276 DataType GetDataTypeFromAttr(const NodeDef& node, const string& type_attr);
/external/tensorflow/tensorflow/python/eager/
Dpywrap_tfe_src.cc81 if (!op_def.input_arg(i).type_attr().empty()) { in GetAttrToInputsMap()
82 auto it = m->find(op_def.input_arg(i).type_attr()); in GetAttrToInputsMap()
84 it = m->insert({op_def.input_arg(i).type_attr(), {}}).first; in GetAttrToInputsMap()
2369 return MaybeGetDTypeForAttr(input_arg.type_attr(), op_exec_info);
2372 op_exec_info->cached_dtypes[input_arg.type_attr()] =
2381 if (add_type_attr && !input_arg.type_attr().empty()) {
2383 TFE_OpSetAttrType(op, input_arg.type_attr().data(), dtype);
2386 GetPythonObjectFromString(input_arg.type_attr().data()));

12