Searched refs:func_instantiation_attr (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/utils/ |
| D | functions_test.cc | 81 protobuf::Map<string, AttrValue> func_instantiation_attr; in TEST_F() local
82 func_instantiation_attr["key"].set_s("key-value"); in TEST_F()
83 func_instantiation_attr["A"].set_type(DT_FLOAT); in TEST_F()
84 func_instantiation_attr["B"].set_type(DT_INT32); in TEST_F()
85 func_instantiation_attr["C"].mutable_list()->add_type(DT_FLOAT); in TEST_F()
86 func_instantiation_attr["C"].mutable_list()->add_type(DT_INT32); in TEST_F()
87 func_instantiation_attr["D"].set_type(DT_DOUBLE); in TEST_F()
91 func, AttrSlice(&func_instantiation_attr), &type_parameters)); in TEST_F()
102 func, AttrSlice(&func_instantiation_attr), &body_parameters)); in TEST_F()
126 protobuf::Map<string, AttrValue> func_instantiation_attr; in TEST_F() local
[all …]
|
| D | functions.h | 135 const FunctionDef& func, const AttrSlice& func_instantiation_attr,
142 const FunctionDef& func, const AttrSlice& func_instantiation_attr,
162 const AttrSlice& func_instantiation_attr,
|
| D | functions.cc | 157 const FunctionDef& func, const AttrSlice& func_instantiation_attr, in InstantiationTypeParameters() argument
167 GetNodeAttr(func_instantiation_attr, arg.type_attr(), &dtype)); in InstantiationTypeParameters()
173 GetNodeAttr(func_instantiation_attr, arg.type_list_attr(), &dtypes)); in InstantiationTypeParameters()
193 const FunctionDef& func, const AttrSlice& func_instantiation_attr, in InstantiationBodyParameters() argument
208 func_instantiation_attr.Find(placeholder); in InstantiationBodyParameters()
222 const AttrSlice& func_instantiation_attr, in MakeGrapplerFunctionItem() argument
244 FunctionDefToBodyHelper(func, func_instantiation_attr, &flib, &fbody)); in MakeGrapplerFunctionItem()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | function_optimizer.cc | 642 const AttrSlice& func_instantiation_attr, in InitializeFunctionSpecializationSignature() argument
651 TF_RETURN_IF_ERROR(InstantiationTypeParameters(func, func_instantiation_attr, in InitializeFunctionSpecializationSignature()
653 TF_RETURN_IF_ERROR(InstantiationBodyParameters(func, func_instantiation_attr, in InitializeFunctionSpecializationSignature()
685 const AttrSlice func_instantiation_attr = in SpecializeFunction() local
690 func_node, func, func_instantiation_attr, *ctx, &signature)); in SpecializeFunction()
720 func, func_instantiation_attr, flib, ctx->graph_version(), &item)); in SpecializeFunction()
|