Searched refs:func_instantiation_attr (Results 1 – 4 of 4) sorted by relevance
/external/tensorflow/tensorflow/core/grappler/utils/ |
D | functions_test.cc | 76 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 …]
|
D | functions.h | 140 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,
|
D | functions.cc | 61 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/ |
D | function_optimizer.cc | 732 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()
|