Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/
H A 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
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/data/
H A 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::__anon76f4007d0111::ReduceDatasetParams
H A 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::__anon708df34b0111::FilterDatasetParams
H A 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::__anon442afc890111::FlatMapDatasetParams
H A 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::__anon15242baf0111::MapDefunOpParams
H A 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::__anon4c6659970111::ParallelFilterDatasetParams
H A 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::__anone2fd5c2f0111::ParallelMapDatasetParams
H A 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::__anonaff64af40111::InterleaveDatasetParams
H A 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::__anonf2f0f7400111::ParallelInterleaveDatasetParams
H A Dmap_defun_op.cc170 auto func_lib = ctx->function_library(); in MapDefunOp() local
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/data/experimental/
H A 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::__anon71359bb30111::MapAndBatchDatasetParams
H A 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
H A 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::__anon0cf95fbd0111::ParallelInterleaveDatasetParams
/aosp_15_r20/external/tensorflow/tensorflow/core/data/
H A Ddataset_test_base.h165 virtual std::vector<FunctionDef> func_lib() const { return {}; } in func_lib() function
263 std::vector<FunctionDef> func_lib, in MapDatasetParams()
H A Ddataset_test_base.cc1091 std::vector<FunctionDef> MapDatasetParams::func_lib() const { in func_lib() function in tensorflow::data::MapDatasetParams
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/eager/
H A Dcontext.h165 FunctionLibraryRuntime* func_lib(const Device* d) const { in func_lib() function
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/
H A Dimport_model.cc1393 const auto& func_lib = graph_flib_; in ConvertLibFunction() local