| /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::__anon3adef36e0111::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::__anonae9e6fba0111::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::__anon3478e63c0111::FilterDatasetParams
|
| 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::__anonfe0e44000111::MapDefunOpParams
|
| D | parallel_map_dataset_op_test.cc | 30 std::vector<FunctionDef> func_lib, DataTypeVector type_arguments, in ParallelMapDatasetParams() argument
40 func_lib_(std::move(func_lib)), in ParallelMapDatasetParams()
88 std::vector<FunctionDef> func_lib() const override { return func_lib_; } in func_lib() function in tensorflow::data::__anon7368ffe00111::ParallelMapDatasetParams
|
| D | map_defun_op.cc | 168 auto func_lib = ctx->function_library(); in MapDefunOp() local
169 OP_REQUIRES(ctx, func_lib != nullptr, in MapDefunOp()
174 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::__anon85134e650111::InterleaveDatasetParams
|
| D | parallel_interleave_dataset_op_test.cc | 31 std::vector<FunctionDef> func_lib, DataTypeVector type_arguments, in ParallelInterleaveDatasetParams() argument
44 func_lib_(std::move(func_lib)), in ParallelInterleaveDatasetParams()
101 std::vector<FunctionDef> func_lib() const override { return func_lib_; } in func_lib() function in tensorflow::data::__anonc10b69d10111::ParallelInterleaveDatasetParams
|
| D | dataset_test_base.h | 157 virtual std::vector<FunctionDef> func_lib() const { return {}; } in func_lib() function
246 std::vector<FunctionDef> func_lib, in MapDatasetParams() argument
255 func_lib_(std::move(func_lib)), in MapDatasetParams()
273 std::vector<FunctionDef> func_lib() const override;
|
| D | dataset_test_base.cc | 788 InitFunctionLibraryRuntime(dataset_params.func_lib(), cpu_num_)); in InitializeRuntime()
1063 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::__anon402561240111::MapAndBatchDatasetParams
|
| D | parallel_interleave_dataset_op_test.cc | 33 std::vector<FunctionDef> func_lib, DataTypeVector type_arguments, in ParallelInterleaveDatasetParams() argument
45 func_lib_(std::move(func_lib)), in ParallelInterleaveDatasetParams()
98 std::vector<FunctionDef> func_lib() const override { return func_lib_; } in func_lib() function in tensorflow::data::experimental::__anon9e5db5ee0111::ParallelInterleaveDatasetParams
|
| /external/tensorflow/tensorflow/core/distributed_runtime/eager/ |
| D | remote_copy_node.cc | 48 FunctionLibraryRuntime* flr = ctx.func_lib(device); in CreateUncachedKernelAndDeviceOp()
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | context.h | 149 FunctionLibraryRuntime* func_lib(const Device* d) const { in func_lib() function
|
| D | execute.cc | 500 device == nullptr ? nullptr : ctx.func_lib(device); in GetOrCreateKernelAndDevice()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/ |
| D | import_model.cc | 1240 const auto& func_lib = graph_flib_; in ConvertLibFunction() local
1241 const auto* func_def = func_lib.Find(std::string(func_name)); in ConvertLibFunction()
1251 FunctionDefToBodyHelper(*func_def, AttrSlice(), &func_lib, &fbody)); in ConvertLibFunction()
1275 auto grad_func_name = func_lib.FindGradient(std::string(func_name)); in ConvertLibFunction()
|