Home
last modified time | relevance | path

Searched defs:arg_attr (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/core/framework/
Dfunction.proto40 map<uint32, ArgAttrs> arg_attr = 7; field
Dfunction.cc205 for (const auto& arg_attr : arg_attrs->attr()) { in BuildInputArgIndex() local
1004 auto arg_attr = iter->second.attr().find("_composite_device"); in GetFunctionResourceInputDevice() local
/external/tensorflow/tensorflow/core/grappler/utils/
Dfunctions.cc41 std::vector<const FunctionDef::ArgAttrs*> arg_attr, in GrapplerFunctionItem()
114 GrapplerFunctionItem::arg_attr() const { in arg_attr() function in tensorflow::grappler::GrapplerFunctionItem
288 std::vector<const FunctionDef::ArgAttrs*> arg_attr(inputs.size(), nullptr); in MakeGrapplerFunctionItem() local
Dfunctions_test.cc529 FunctionDef::ArgAttrs arg_attr; in TEST_F() local
/external/tensorflow/tensorflow/core/kernels/data/
Dcaptured_function.cc641 auto arg_attr = iter->second.attr().find("_composite_device"); in Instantiate() local
/external/tensorflow/tensorflow/compiler/mlir/lite/
Dflatbuffer_export.cc1557 for (const auto& arg_attr : dict_attrs) { in GetStringsFromDictionaryAttr() local