Searched refs:PopulateFunction (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | function_api_info_test.cc | 47 void PopulateFunction(const string& name, const string& api_interface_name, in PopulateFunction() function
76 PopulateFunction("DoStuffCpu", "DoStuff", "CPU", func_args, output_args, "", in PopulateSampleLibrary()
78 PopulateFunction("DoStuffGpu", "DoStuff", "GPU", in PopulateSampleLibrary()
81 PopulateFunction("DoThings", "DoThings", "", func_args, output_args, "", "", in PopulateSampleLibrary()
83 PopulateFunction("OneOff", "", "", func_args, output_args, "", "", in PopulateSampleLibrary()
85 PopulateFunction("AnotherOneOff", "", "", func_args, output_args, "", "", in PopulateSampleLibrary()
95 PopulateFunction("DoStuffCpu", "DoStuff", "CPU", input_args, output_args, "", in PopulateComplexLibrary()
97 PopulateFunction("DoStuffCpu_gradient", "DoStuff", "CPU", output_args, in PopulateComplexLibrary()
99 PopulateFunction("DoStuffGpu", "DoStuff", "GPU", input_args, in PopulateComplexLibrary()
102 PopulateFunction("DoStuffGpu_gradient", "DoStuff", "GPU", output_with_state, in PopulateComplexLibrary()
|