Searched refs:arg_attrs (Results 1 – 10 of 10) sorted by relevance
| /external/tensorflow/tensorflow/python/framework/ |
| D | function_def_to_graph.py | 182 arg_attrs = fdef.arg_attr[i].attr
183 for k in arg_attrs:
187 node_def.attr["shape"].shape.CopyFrom(arg_attrs[k].list.shape[0])
189 node_def.attr[k].CopyFrom(arg_attrs[k])
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | graph_to_functiondef.cc | 429 FunctionDef::ArgAttrs arg_attrs; in GraphToFunctionDef() local
437 arg_attrs.mutable_attr()->insert(attr); in GraphToFunctionDef()
445 arg_attrs.mutable_attr()->insert({"_output_shapes", value}); in GraphToFunctionDef()
453 arg_attrs.mutable_attr()->insert({"_output_shapes", value}); in GraphToFunctionDef()
459 if (arg_attrs.attr_size() > 0) { in GraphToFunctionDef()
460 (*fdef->mutable_arg_attr())[i] = std::move(arg_attrs); in GraphToFunctionDef()
|
| D | function_test.cc | 1513 FunctionDef::ArgAttrs arg_attrs; in TEST() local
1514 (*arg_attrs.mutable_attr())["_output_shapes"] = std::move(shape_attr); in TEST()
1515 (*fdef.mutable_arg_attr())[0] = std::move(arg_attrs); in TEST()
1574 FunctionDef::ArgAttrs arg_attrs; in TEST() local
1575 *(*arg_attrs.mutable_attr())["_composite_device"].mutable_s() = in TEST()
1577 (*fdef.mutable_arg_attr())[0] = arg_attrs; in TEST()
|
| D | function.cc | 173 const FunctionDef::ArgAttrs* arg_attrs, in BuildInputArgIndex() argument
204 if (arg_attrs) { in BuildInputArgIndex()
205 for (const auto& arg_attr : arg_attrs->attr()) { in BuildInputArgIndex()
735 const FunctionDef::ArgAttrs* arg_attrs = in InstantiateFunction() local
742 s = helper.BuildInputArgIndex(arg_def, attr_values, arg_attrs, in InstantiateFunction()
|
| /external/tensorflow/tensorflow/compiler/mlir/tfr/passes/ |
| D | raise_to_tf.cc | 102 SmallVectorImpl<Value>* inputs, NamedAttrList* arg_attrs,
227 SmallVectorImpl<Value>* inputs, NamedAttrList* arg_attrs, in CollectInputsAndAttributes() argument
237 arg_attrs->push_back( in CollectInputsAndAttributes()
301 arg_attrs->push_back(rewriter.getNamedAttr(attr_name.getValue(), value)); in CollectInputsAndAttributes()
|
| /external/tensorflow/tensorflow/c/ |
| D | c_api_function_test.cc | 1312 auto arg_attrs = func_->fdef.arg_attr().find(0); in TEST_F() local
1313 ASSERT_NE(arg_attrs, func_->fdef.arg_attr().end()); in TEST_F()
1314 auto iter = arg_attrs->second.attr().find("_test_attr"); in TEST_F()
1315 ASSERT_NE(iter, arg_attrs->second.attr().end()); in TEST_F()
|
| /external/python/cpython3/Lib/ |
| D | typing.py | 1523 arg_attrs = arg.__dict__.copy()
1526 arg_attrs.pop(attr)
1527 for obj in arg_attrs.values():
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/ |
| D | flatbuffer_export.cc | 1580 llvm::SmallVector<mlir::DictionaryAttr, 4> arg_attrs, res_attrs; in BuildSignaturedef() local
1581 main_op.getAllArgAttrs(arg_attrs); in BuildSignaturedef()
1584 GetStringsFromDictionaryAttr(arg_attrs, kSignatureDefIndexPath); in BuildSignaturedef()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/ |
| D | import_model.cc | 1485 llvm::SmallVector<mlir::NamedAttrList, 4> arg_attrs; in ConvertFunctionArgAndRets() local
1486 arg_attrs.resize(func.getNumArguments()); in ConvertFunctionArgAndRets()
1505 arg_attrs[index].set(dialect_attribute, converted_attr); in ConvertFunctionArgAndRets()
1539 arg_attrs[i].set("tf.device", builder_.getStringAttr( in ConvertFunctionArgAndRets()
1597 llvm::to_vector<4>(llvm::map_range(arg_attrs, [&](NamedAttrList& list) { in ConvertFunctionArgAndRets()
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | function_test.py | 241 arg_attrs = read_var.get_concrete_function().function_def.arg_attr
242 self.assertLen(arg_attrs, 2)
243 self.assertEqual(arg_attrs[0].attr['_composite_device'].s,
245 self.assertEqual(arg_attrs[1].attr['_composite_device'].s,
|