Home
last modified time | relevance | path

Searched defs:func_lib (Results 1 – 17 of 17) sorted by relevance

/external/tensorflow/tensorflow/core/grappler/optimizers/
Dfunction_api_info_test.cc71 FunctionDefLibrary* func_lib) { in PopulateSampleLibrary()
89 void PopulateComplexLibrary(FunctionDefLibrary* func_lib) { in PopulateComplexLibrary()
134 FunctionDefLibrary func_lib; in TEST() local
156 FunctionDefLibrary func_lib; in TEST() local
181 FunctionDefLibrary func_lib; in TEST() local
/external/tensorflow/tensorflow/core/kernels/data/
Dreduce_dataset_op_test.cc30 std::vector<FunctionDef> func_lib, in ReduceDatasetParams()
83 std::vector<FunctionDef> func_lib() const override { return func_lib_; } in func_lib() function in tensorflow::data::__anonac3e1e8c0111::ReduceDatasetParams
Dfilter_dataset_op_test.cc28 std::vector<FunctionDef> func_lib, in FilterDatasetParams()
71 std::vector<FunctionDef> func_lib() const override { return func_lib_; } in func_lib() function in tensorflow::data::__anona5d8115a0111::FilterDatasetParams
Dflat_map_dataset_op_test.cc28 std::vector<FunctionDef> func_lib, in FlatMapDatasetParams()
71 std::vector<FunctionDef> func_lib() const override { return func_lib_; } in func_lib() function in tensorflow::data::__anonf474da580111::FlatMapDatasetParams
Dmap_defun_op_test.cc31 std::vector<FunctionDef> func_lib, in MapDefunOpParams()
76 std::vector<FunctionDef> func_lib() const override { return func_lib_; } in func_lib() function in tensorflow::data::__anon8ac0845e0111::MapDefunOpParams
Dparallel_filter_dataset_op_test.cc35 std::vector<FunctionDef> func_lib, DataTypeVector type_arguments, in ParallelFilterDatasetParams()
84 std::vector<FunctionDef> func_lib() const override { return func_lib_; } in func_lib() function in tensorflow::data::__anon8e90b8860111::ParallelFilterDatasetParams
Dparallel_map_dataset_op_test.cc30 std::vector<FunctionDef> func_lib, DataTypeVector type_arguments, in ParallelMapDatasetParams()
87 std::vector<FunctionDef> func_lib() const override { return func_lib_; } in func_lib() function in tensorflow::data::__anon3d8f897e0111::ParallelMapDatasetParams
Dinterleave_dataset_op_test.cc29 std::vector<FunctionDef> func_lib, in InterleaveDatasetParams()
80 std::vector<FunctionDef> func_lib() const override { return func_lib_; } in func_lib() function in tensorflow::data::__anon9a2ed8830111::InterleaveDatasetParams
Dparallel_interleave_dataset_op_test.cc36 std::vector<FunctionDef> func_lib, DataTypeVector type_arguments, in ParallelInterleaveDatasetParams()
106 std::vector<FunctionDef> func_lib() const override { return func_lib_; } in func_lib() function in tensorflow::data::__anonefff35af0111::ParallelInterleaveDatasetParams
Dmap_defun_op.cc170 auto func_lib = ctx->function_library(); in MapDefunOp() local
/external/tensorflow/tensorflow/core/kernels/data/experimental/
Dmap_and_batch_dataset_op_test.cc30 std::vector<FunctionDef> func_lib, DataTypeVector type_arguments, in MapAndBatchDatasetParams()
83 std::vector<FunctionDef> func_lib() const override { return func_lib_; } in func_lib() function in tensorflow::data::experimental::__anond49d67420111::MapAndBatchDatasetParams
Dio_ops_test.cc40 std::vector<FunctionDef> func_lib, bool use_shard_func, in SaveDatasetV2Params()
89 std::vector<FunctionDef> func_lib() const override { return func_lib_; } in func_lib() function in tensorflow::data::experimental::SaveDatasetV2Params
Dparallel_interleave_dataset_op_test.cc33 std::vector<FunctionDef> func_lib, DataTypeVector type_arguments, in ParallelInterleaveDatasetParams()
98 std::vector<FunctionDef> func_lib() const override { return func_lib_; } in func_lib() function in tensorflow::data::experimental::__anon8f955a8c0111::ParallelInterleaveDatasetParams
/external/tensorflow/tensorflow/core/data/
Ddataset_test_base.h165 virtual std::vector<FunctionDef> func_lib() const { return {}; } in func_lib() function
263 std::vector<FunctionDef> func_lib, in MapDatasetParams()
Ddataset_test_base.cc1091 std::vector<FunctionDef> MapDatasetParams::func_lib() const { in func_lib() function in tensorflow::data::MapDatasetParams
/external/tensorflow/tensorflow/core/common_runtime/eager/
Dcontext.h165 FunctionLibraryRuntime* func_lib(const Device* d) const { in func_lib() function
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/
Dimport_model.cc1393 const auto& func_lib = graph_flib_; in ConvertLibFunction() local