Searched refs:AddToFunctionLibrary (Results 1 – 6 of 6) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/core/data/ |
| H A D | dataset_utils_test.cc | 50 TEST(DatasetUtilsTest, AddToFunctionLibrary) { in TEST() argument
84 TF_ASSERT_OK(AddToFunctionLibrary(&flib_0, fdef_to_add)); in TEST()
88 TF_ASSERT_OK(AddToFunctionLibrary(&flib_1, flib_to_add)); in TEST()
118 Status s = AddToFunctionLibrary(&flib_0, fdef_to_add); in TEST()
127 s = AddToFunctionLibrary(&flib_1, flib_to_add); in TEST()
|
| H A D | dataset_utils.h | 198 Status AddToFunctionLibrary(FunctionLibraryDefinition* base,
200 Status AddToFunctionLibrary(FunctionLibraryDefinition* base,
|
| H A D | dataset_utils.cc | 303 Status AddToFunctionLibrary(FunctionLibraryDefinition* base, in AddToFunctionLibrary() function
318 Status AddToFunctionLibrary(FunctionLibraryDefinition* base, in AddToFunctionLibrary() function
|
| H A D | rewrite_utils.cc | 195 TF_RETURN_IF_ERROR(AddToFunctionLibrary(lib_def.get(), graph_def.library())); in RewriteDataset()
|
| H A D | serialization_utils.cc | 64 TF_RETURN_IF_ERROR(AddToFunctionLibrary(lib_def.get(), graph_def.library())); in FromGraphDef()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/data/ |
| H A D | dataset_ops.cc | 170 AddToFunctionLibrary(flib_def.get(), graph_def.library())); in Compute()
|