| /aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/data/ |
| H A D | split_utils_test.cc | 188 FunctionLibraryDefinition function_library(OpRegistry::Global()); in TEST() local
233 FunctionLibraryDefinition function_library(OpRegistry::Global()); in TEST() local
262 FunctionLibraryDefinition function_library(OpRegistry::Global()); in TEST() local
296 FunctionLibraryDefinition function_library(OpRegistry::Global()); in TEST() local
330 FunctionLibraryDefinition function_library(OpRegistry::Global()); in TEST() local
372 FunctionLibraryDefinition function_library(OpRegistry::Global()); in TEST() local
399 FunctionLibraryDefinition function_library(OpRegistry::Global()); in TEST() local
430 FunctionLibraryDefinition function_library(OpRegistry::Global()); in TEST() local
472 FunctionLibraryDefinition function_library(OpRegistry::Global()); in TEST() local
|
| H A D | filter_parallelization.cc | 74 FunctionLibraryDefinition function_library(OpRegistry::Global(), in OptimizeAndCollectStats() local
|
| H A D | batch_parallelization.cc | 76 FunctionLibraryDefinition function_library(OpRegistry::Global(), in OptimizeAndCollectStats() local
|
| H A D | map_parallelization.cc | 75 FunctionLibraryDefinition function_library(OpRegistry::Global(), in OptimizeAndCollectStats() local
|
| H A D | filter_fusion.cc | 73 FunctionLibraryDefinition function_library(OpRegistry::Global(), in OptimizeAndCollectStats() local
|
| H A D | map_fusion.cc | 92 FunctionLibraryDefinition function_library(OpRegistry::Global(), in OptimizeAndCollectStats() local
|
| H A D | noop_elimination.cc | 120 FunctionLibraryDefinition function_library(OpRegistry::Global(), in IsMapIdentity() local
|
| H A D | map_and_filter_fusion.cc | 173 FunctionLibraryDefinition function_library(OpRegistry::Global(), in OptimizeAndCollectStats() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/ |
| H A D | function_optimizer_test.cc | 1436 std::vector<FunctionDef> function_library = {x_times_two}; in TEST_F() local
1485 std::vector<FunctionDef> function_library = {x_times_two}; in TEST_F() local
1553 std::vector<FunctionDef> function_library = {mul_func}; in TEST_F() local
1617 std::vector<FunctionDef> function_library = {mul_func}; in TEST_F() local
1695 std::vector<FunctionDef> function_library = {mul_func}; in TEST_F() local
1819 std::vector<FunctionDef> function_library = {my_func}; in TEST_F() local
1927 std::vector<FunctionDef> function_library = {my_func}; in TEST_F() local
|
| H A D | function_api_info.cc | 157 const FunctionDefLibrary& function_library) { in Init()
|
| H A D | function_optimizer.cc | 274 const FunctionLibraryDefinition& function_library() const { in function_library() function in tensorflow::grappler::__anon7d4353b00111::FunctionOptimizerContext
277 FunctionLibraryDefinition& function_library() { return function_library_; } in function_library() function in tensorflow::grappler::__anon7d4353b00111::FunctionOptimizerContext
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tf2xla/ |
| H A D | mlir_bridge_rollout_policy.cc | 24 const FunctionLibraryDefinition* function_library, in GetMlirBridgeRolloutPolicy()
41 const FunctionLibraryDefinition* function_library, in LogGraphFeatures()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/grappler/verifiers/ |
| H A D | structure_verifier.cc | 37 FunctionLibraryDefinition function_library(OpRegistry::Global(), in Verify() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| H A D | graph_optimization_pass.h | 34 const tensorflow::FunctionLibraryDefinition& function_library) in GetPassState()
|
| H A D | graph_optimization_pass.cc | 37 const tensorflow::FunctionLibraryDefinition& function_library) { in Run()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/framework/ |
| H A D | op_kernel.h | 316 FunctionLibraryRuntime* function_library() const { return flib_; } in function_library() function
657 FunctionLibraryRuntime* function_library = nullptr; member
1087 FunctionLibraryRuntime* function_library() const { in function_library() function
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/grappler/costs/ |
| H A D | op_context.h | 32 const FunctionDefLibrary* function_library; // Not owned. member
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tfr/integration/ |
| H A D | graph_decompose_pass.cc | 44 const FunctionLibraryDefinition& function_library) { in Run()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/ |
| H A D | xla_op_kernel.h | 308 FunctionLibraryRuntime* function_library() const { in function_library() function
|
| H A D | mlir_bridge_pass.cc | 218 const FunctionLibraryDefinition& function_library) { in Run()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/ |
| H A D | local_executor_params.h | 41 FunctionLibraryRuntime* function_library = nullptr; member
|
| H A D | graph_runner.cc | 103 Status GraphRunner::Run(Graph* graph, FunctionLibraryRuntime* function_library, in Run()
|
| /aosp_15_r20/external/tensorflow/tensorflow/cc/experimental/libexport/ |
| H A D | load.cc | 125 auto& function_library = in GetFunctionDefs() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/jit/ |
| H A D | xla_platform_info.cc | 169 const FunctionLibraryRuntime& function_library, DeviceBase* device, in GenerateCompilerOptions()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/ |
| H A D | mlir_graph_optimization_pass_test.cc | 75 const FunctionLibraryDefinition& function_library) override { in Run()
|