Home
last modified time | relevance | path

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

/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"
76 type_attr: "T"
119 type_attr: "T"
124 type_attr: "T"
168 type_attr: "T"
173 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
227 if arg.type_attr:
228 attr_type = _Attr(op_def, arg.type_attr).type
232 (arg.type_attr, op_def.name, attr_type))
238 (arg.type_attr, op_def.name, attr_type))
434 if input_arg.type_attr in attrs:
435 dtype = attrs[input_arg.type_attr]
444 if dtype is None and input_arg.type_attr in default_type_attr_map:
445 default_dtype = default_type_attr_map[input_arg.type_attr]
478 elif input_arg.type_attr in attrs:
[all …]
Dpython_op_gen.cc189 original_arg = gtl::FindOrNull(inferred_attrs, arg.type_attr()); in ArgTypeName()
195 arg.type_attr(), "`."); in ArgTypeName()
197 const OpDef::AttrDef* attr = FindAttr(arg.type_attr(), op_def); in ArgTypeName()
212 } else if (!arg.type_attr().empty() || !arg.type_list_attr().empty()) { in ArgTypeName()
214 const string attr_name = is_list ? arg.type_list_attr() : arg.type_attr(); in ArgTypeName()
490 if (!arg.type_attr().empty()) { in Code()
491 gtl::InsertIfNotPresent(&inferred_attrs_, arg.type_attr(), arg.name()); in Code()
Dimporter.py53 if arg_def.type_attr:
54 dtype = _GetNodeAttr(node_def, arg_def.type_attr).type
59 elif arg_def.type_attr:
60 return [_GetNodeAttr(node_def, arg_def.type_attr).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/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/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.cc39 return arg.type() != DT_INVALID || !arg.type_attr().empty() || in HasAttrStyleType()
202 (!arg.type_attr().empty() ? 1 : 0) == in ValidateArg()
207 (!arg.type_attr().empty() ? 1 : 0) + in ValidateArg()
214 if (!arg.type_attr().empty()) { in ValidateArg()
215 const OpDef::AttrDef* attr = FindAttr(arg.type_attr(), op_def); in ValidateArg()
216 VALIDATE(attr != nullptr, "No attr with name '", arg.type_attr(), "'", in ValidateArg()
366 strings::StrAppend(&ret, arg.type_attr()); in SummarizeArgs()
564 gtl::FindPtrOrNull(old_attrs, arg.type_attr()); in ComputeArgSignature()
567 strings::StrAppend(&type, arg.type_attr()); in ComputeArgSignature()
572 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.cc307 if (!arg_def.type_attr().empty()) { in AddArgToSig()
309 TF_RETURN_IF_ERROR(GetNodeAttr(node_def, arg_def.type_attr(), &dtype)); in AddArgToSig()
321 } else if (!arg_def.type_attr().empty()) { in AddArgToSig()
324 AttrSlice(node_def).Find(arg_def.type_attr(), &attr_value)); in AddArgToSig()
493 } else if (!arg_def.type_attr().empty() || arg_def.type() != DT_INVALID) { in ComputeArgRange()
Dnode_def_builder.cc123 Attr(input_arg->type_attr(), BaseType(dt)); in SingleInput()
142 Attr(input_arg->type_attr(), base); in ListInput()
Dfunction.cc71 return errors::NotFound("type attr not found: ", arg_def.type_attr()); in ArgNumType()
79 } else if (arg_def.type_attr().empty()) { in ArgNumType()
82 const AttrValue* v = attrs.Find(arg_def.type_attr()); in ArgNumType()
84 return errors::NotFound("type attr not found: ", arg_def.type_attr()); in ArgNumType()
455 strings::StrAppend(&out, arg.type_attr()); in Print()
Dop_def_builder_test.cc582 type_attr: "dtype" in TEST_F()
/external/tensorflow/tensorflow/compiler/tf2xla/
Dxla_op_registry.cc205 for (const string& type_attr : type_attrs) { in RegisterCompilationKernels() local
207 attr_constraint->set_name(type_attr); in RegisterCompilationKernels()
211 const OpDef::AttrDef& op_def_attr = *FindAttr(type_attr, *op_def); in RegisterCompilationKernels()
216 auto constraint_it = op_registration->type_constraints.find(type_attr); in RegisterCompilationKernels()
/external/tensorflow/tensorflow/c/eager/
Dc_api.cc664 const string& type_attr = op_def_arg.type_attr(); in OpToFunction() local
665 if (!type_attr.empty()) { in OpToFunction()
666 auto i = ndef.attr().find(type_attr); in OpToFunction()
669 tensorflow::strings::StrCat("Could not find attr ", type_attr, in OpToFunction()
/external/tensorflow/tensorflow/core/grappler/
Dgrappler_item_builder.cc534 auto it = func_attr.find(inp.type_attr()); in GrapplerItemFromFunctionDef()
536 LOG(ERROR) << "Unknown type attribute " << inp.type_attr() in GrapplerItemFromFunctionDef()
/external/tensorflow/tensorflow/python/eager/
Dpython_eager_op_gen.cc248 if (!arg.type_attr().empty()) { in Code()
249 AddAttrForArg(arg.type_attr(), i); in Code()
862 if (!arg.type_attr().empty() || !arg.type_list_attr().empty()) continue; in AddEagerInputCasts()
Dpywrap_tfe_src.cc1375 if (input_arg != nullptr && !input_arg->type_attr().empty()) {
1377 TFE_OpSetAttrType(op, input_arg->type_attr().data(), dtype);
1380 GetPythonObjectFromString(input_arg->type_attr().data()));
/external/python/cpython3/Modules/_ctypes/
D_ctypes.c1320 PyObject *length_attr, *type_attr; in PyCArrayType_new() local
1334 type_attr = NULL; in PyCArrayType_new()
1353 type_attr = PyObject_GetAttrString((PyObject *)result, "_type_"); in PyCArrayType_new()
1354 if (!type_attr) { in PyCArrayType_new()
1365 itemdict = PyType_stgdict(type_attr); in PyCArrayType_new()
1403 stgdict->proto = type_attr; in PyCArrayType_new()
1432 Py_XDECREF(type_attr); in PyCArrayType_new()
/external/tensorflow/tensorflow/cc/framework/
Dcc_op_gen.cc447 if (!arg.type_attr().empty()) { in InferArgAttributes()
448 gtl::InsertIfNotPresent(inferred_attrs, arg.type_attr(), arg.name()); in InferArgAttributes()