Home
last modified time | relevance | path

Searched defs:func_attr (Results 1 – 10 of 10) sorted by relevance

/external/tensorflow/tensorflow/core/grappler/optimizers/
Dimplementation_selector_test.cc63 auto* func_attr = cpu_def.mutable_attr(); in TEST_F() local
101 auto* func_attr = cpu_def.mutable_attr(); in TEST_F() local
Dfunction_api_info_test.cc59 auto* func_attr = func_def->mutable_attr(); in PopulateFunction() local
Dfunction_optimizer.cc112 auto* func_attr = AttrSlice(func_node).Find(kFuncAttrName); in IsIndirectFunctionCall() local
123 auto* func_attr = AttrSlice(func_node).Find(kFuncAttrName); in FunctionInstantiationAttributes() local
479 const AttrValue* func_attr = AttrSlice(node).Find("f"); in FindFunctionCall() local
/external/tensorflow/tensorflow/compiler/tf2xla/
Dfunctionalize_control_flow.cc261 const string func_attr = it->second; in Run() local
Dtf2xla_util.cc180 NameAttrList func_attr; in PropagateConstIntoFuncAttr() local
/external/tensorflow/tensorflow/core/grappler/optimizers/data/
Dmap_vectorization.cc281 auto& func_attr = (*map_node.mutable_attr())["f"]; in AddNewMapNode() local
386 NameAttrList func_attr; in AddBranch() local
Dvectorization_utils.cc627 AttrValue func_attr; in GetResult() local
Dvectorization_utils_test.cc71 NameAttrList func_attr; in WrapFunctionWithMapDefun() local
/external/tensorflow/tensorflow/core/grappler/utils/
Dfunctions.cc311 string func_name, string description, AttrSlice func_attr, in GrapplerFunctionItem()
383 const AttrSlice& GrapplerFunctionItem::func_attr() const { return func_attr_; } in func_attr() function in tensorflow::grappler::GrapplerFunctionItem
/external/tensorflow/tensorflow/c/
Dc_api_function.cc166 for (const auto& func_attr : fdef->signature().attr()) { in FillFunctionBody() local