Home
last modified time | relevance | path

Searched refs:func_instantiation_attr (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/core/grappler/utils/
Dfunctions_test.cc76 protobuf::Map<string, AttrValue> func_instantiation_attr; in TEST_F() local
77 func_instantiation_attr["key"].set_s("key-value"); in TEST_F()
78 func_instantiation_attr["A"].set_type(DT_FLOAT); in TEST_F()
79 func_instantiation_attr["B"].set_type(DT_INT32); in TEST_F()
80 func_instantiation_attr["C"].set_type(DT_DOUBLE); in TEST_F()
84 func, AttrSlice(&func_instantiation_attr), &type_parameters)); in TEST_F()
93 func, AttrSlice(&func_instantiation_attr), &body_parameters)); in TEST_F()
242 protobuf::Map<string, AttrValue> func_instantiation_attr; in TEST_F() local
243 func_instantiation_attr["T"].set_type(DT_FLOAT); in TEST_F()
248 AttrSlice(&func_instantiation_attr), in TEST_F()
[all …]
Dfunctions.h140 explicit GrapplerFunctionItemInstantiation(AttrSlice func_instantiation_attr) in GrapplerFunctionItemInstantiation() argument
141 : func_instantiation_attr_(func_instantiation_attr) {} in GrapplerFunctionItemInstantiation()
225 const FunctionDef& func, const AttrSlice& func_instantiation_attr,
232 const FunctionDef& func, const AttrSlice& func_instantiation_attr,
261 const AttrSlice& func_instantiation_attr,
Dfunctions.cc61 const AttrSlice& func_instantiation_attr, NodeDef* node) { in ResolveFunctionBodyNodeAttrPlaceholders() argument
66 const AttrValue* attr_value = func_instantiation_attr.Find(placeholder); in ResolveFunctionBodyNodeAttrPlaceholders()
424 const FunctionDef& func, const AttrSlice& func_instantiation_attr, in InstantiationTypeParameters() argument
430 GrapplerFunctionItemInstantiation instantiation(func_instantiation_attr); in InstantiationTypeParameters()
452 const FunctionDef& func, const AttrSlice& func_instantiation_attr, in InstantiationBodyParameters() argument
468 func_instantiation_attr.Find(placeholder); in InstantiationBodyParameters()
482 const AttrSlice& func_instantiation_attr, in MakeGrapplerFunctionItem() argument
502 GrapplerFunctionItemInstantiation instantiation(func_instantiation_attr); in MakeGrapplerFunctionItem()
589 func_instantiation_attr, new_node)); in MakeGrapplerFunctionItem()
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dfunction_optimizer.cc732 const AttrSlice& func_instantiation_attr, in InitializeFunctionSpecializationSignature() argument
741 TF_RETURN_IF_ERROR(InstantiationTypeParameters(func, func_instantiation_attr, in InitializeFunctionSpecializationSignature()
743 TF_RETURN_IF_ERROR(InstantiationBodyParameters(func, func_instantiation_attr, in InitializeFunctionSpecializationSignature()
775 const AttrSlice func_instantiation_attr = in SpecializeFunction() local
780 func_node, func, func_instantiation_attr, *ctx, &signature)); in SpecializeFunction()
810 func, func_instantiation_attr, flib, ctx->graph_version(), &item)); in SpecializeFunction()
1017 const AttrSlice func_instantiation_attr = in InlineDirectFunctionCall() local
1021 Status item_status = MakeGrapplerFunctionItem(func, func_instantiation_attr, in InlineDirectFunctionCall()
1567 const AttrSlice func_instantiation_attr = in InlineIndirectFunctionCall() local
1571 Status item_status = MakeGrapplerFunctionItem(func, func_instantiation_attr, in InlineIndirectFunctionCall()