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::__anone3963afd0111::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::__anondd302dcb0111::FilterDatasetParams
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::__anon9257d62f0111::MapDefunOpParams
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::__anon6245d7090111::FlatMapDatasetParams
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::__anonf00176af0111::ParallelMapDatasetParams
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::__anon797264170111::ParallelFilterDatasetParams
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::__anonc031c5740111::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::__anon83fe41c00111::ParallelInterleaveDatasetParams
Dmap_defun_op.cc170 auto func_lib = ctx->function_library(); in MapDefunOp() local
/external/tensorflow/tensorflow/core/kernels/data/experimental/
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
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::__anonda1616330111::MapAndBatchDatasetParams
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::__anon5bebba3d0111::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