| /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 | reduce_dataset_op_test.cc | 30 std::vector<FunctionDef> func_lib, in ReduceDatasetParams() argument
40 func_lib_(std::move(func_lib)), in ReduceDatasetParams()
82 std::vector<FunctionDef> func_lib() const override { return func_lib_; } in func_lib() function in tensorflow::data::__anon161764120111::ReduceDatasetParams
|
| D | flat_map_dataset_op_test.cc | 28 std::vector<FunctionDef> func_lib, in FlatMapDatasetParams() argument
37 func_lib_(std::move(func_lib)), in FlatMapDatasetParams()
70 std::vector<FunctionDef> func_lib() const override { return func_lib_; } in func_lib() function in tensorflow::data::__anon39b5995e0111::FlatMapDatasetParams
|
| D | filter_dataset_op_test.cc | 28 std::vector<FunctionDef> func_lib, in FilterDatasetParams() argument
37 func_lib_(std::move(func_lib)), in FilterDatasetParams()
70 std::vector<FunctionDef> func_lib() const override { return func_lib_; } in func_lib() function in tensorflow::data::__anon0fb156e00111::FilterDatasetParams
|
| D | parallel_map_dataset_op_test.cc | 29 std::vector<FunctionDef> func_lib, in ParallelMapDatasetParams() argument
40 func_lib_(std::move(func_lib)), in ParallelMapDatasetParams()
85 std::vector<FunctionDef> func_lib() const override { return func_lib_; } in func_lib() function in tensorflow::data::__anona3845b840111::ParallelMapDatasetParams
|
| D | map_defun_op_test.cc | 31 std::vector<FunctionDef> func_lib, in MapDefunOpParams() argument
40 func_lib_(std::move(func_lib)), in MapDefunOpParams()
76 std::vector<FunctionDef> func_lib() const override { return func_lib_; } in func_lib() function in tensorflow::data::__anonbab0be240111::MapDefunOpParams
|
| D | map_defun_op.cc | 158 auto func_lib = ctx->function_library(); in MapDefunOp() local
159 OP_REQUIRES(ctx, func_lib != nullptr, in MapDefunOp()
164 func_lib->Instantiate(func->name(), AttrSlice(&func->attr()), in MapDefunOp()
|
| D | interleave_dataset_op_test.cc | 29 std::vector<FunctionDef> func_lib, in InterleaveDatasetParams() argument
40 func_lib_(std::move(func_lib)), in InterleaveDatasetParams()
79 std::vector<FunctionDef> func_lib() const override { return func_lib_; } in func_lib() function in tensorflow::data::__anon329b71090111::InterleaveDatasetParams
|
| D | parallel_interleave_dataset_op_test.cc | 30 std::vector<FunctionDef> func_lib, DataTypeVector type_arguments, in ParallelInterleaveDatasetParams() argument
41 func_lib_(std::move(func_lib)), in ParallelInterleaveDatasetParams()
90 std::vector<FunctionDef> func_lib() const override { return func_lib_; } in func_lib() function in tensorflow::data::__anon259ec4f50111::ParallelInterleaveDatasetParams
|
| D | dataset_test_base.h | 160 virtual std::vector<FunctionDef> func_lib() const { return {}; } in func_lib() function
249 std::vector<FunctionDef> func_lib, in MapDatasetParams() argument
258 func_lib_(std::move(func_lib)), in MapDatasetParams()
276 std::vector<FunctionDef> func_lib() const override;
|
| D | dataset_test_base.cc | 712 InitFunctionLibraryRuntime(dataset_params.func_lib(), cpu_num_)); in InitializeRuntime()
975 std::vector<FunctionDef> MapDatasetParams::func_lib() const { in func_lib() function in tensorflow::data::MapDatasetParams
|
| /external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| D | map_and_batch_dataset_op_test.cc | 30 std::vector<FunctionDef> func_lib, DataTypeVector type_arguments, in MapAndBatchDatasetParams() argument
40 func_lib_(std::move(func_lib)), in MapAndBatchDatasetParams()
83 std::vector<FunctionDef> func_lib() const override { return func_lib_; } in func_lib() function in tensorflow::data::experimental::__anon01464bc80111::MapAndBatchDatasetParams
|
| D | parallel_interleave_dataset_op_test.cc | 32 std::vector<FunctionDef> func_lib, DataTypeVector type_arguments, in ParallelInterleaveDatasetParams() argument
44 func_lib_(std::move(func_lib)), in ParallelInterleaveDatasetParams()
95 std::vector<FunctionDef> func_lib() const override { return func_lib_; } in func_lib() function in tensorflow::data::experimental::__anonc3550b920111::ParallelInterleaveDatasetParams
|
| /external/tensorflow/tensorflow/core/distributed_runtime/eager/ |
| D | remote_copy_node.cc | 46 FunctionLibraryRuntime* flr = ctx.func_lib(device); in CreateUncachedKernelAndDeviceOp()
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | context.h | 133 FunctionLibraryRuntime* func_lib(const Device* d) const { in func_lib() function
|
| D | execute.cc | 479 device == nullptr ? nullptr : ctx.func_lib(device); in EagerLocalExecute()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/ |
| D | import_model.cc | 1060 const auto& func_lib = graph_flib_; in ConvertLibFunction() local
1061 const auto* func_def = func_lib.Find(std::string(func_name)); in ConvertLibFunction()
1071 FunctionDefToBodyHelper(*func_def, AttrSlice(), &func_lib, &fbody)); in ConvertLibFunction()
1095 auto grad_func_name = func_lib.FindGradient(std::string(func_name)); in ConvertLibFunction()
|