Searched refs:func_lib (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | function_api_info_test.cc | 71 FunctionDefLibrary* func_lib) { in PopulateSampleLibrary() argument
77 "", func_lib->add_function()); in PopulateSampleLibrary()
80 "", func_lib->add_function()); in PopulateSampleLibrary()
82 func_lib->add_function()); in PopulateSampleLibrary()
84 func_lib->add_function()); in PopulateSampleLibrary()
86 func_lib->add_function()); in PopulateSampleLibrary()
89 void PopulateComplexLibrary(FunctionDefLibrary* func_lib) { in PopulateComplexLibrary() argument
96 "DoStuffCpu_gradient", func_lib->add_function()); in PopulateComplexLibrary()
98 input_args, "DoStuffCpu", "", func_lib->add_function()); in PopulateComplexLibrary()
101 func_lib->add_function()); in PopulateComplexLibrary()
[all …]
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | map_dataset_op_test.cc | 99 func_lib(std::move(input_func_lib)) {} in GetNextTestParams()
106 std::vector<FunctionDef> func_lib; member
117 TF_ASSERT_OK(InitFunctionLibraryRuntime(test_params.func_lib, cpu_num)); in TEST_P()
455 func_lib(std::move(input_func_lib)) {} in RoundtripTestParams()
463 std::vector<FunctionDef> func_lib; member
475 TF_ASSERT_OK(InitFunctionLibraryRuntime(test_params.func_lib, cpu_num)); in TEST_P()
|
| D | map_defun_op.cc | 47 auto func_lib = ctx->function_library(); in MapDefunOp() local
48 OP_REQUIRES(ctx, func_lib != nullptr, in MapDefunOp()
53 func_lib->Instantiate(func->name(), AttrSlice(&func->attr()), in MapDefunOp()
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | context.h | 92 FunctionLibraryRuntime* func_lib(Device* d) const { in func_lib() function
|
| D | execute.cc | 420 device == nullptr ? nullptr : ctx->func_lib(device); in EagerLocalExecute()
|