/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 205 SplitFunction(orig, {"i1"}, /*num_captured_inputs=*/0, function_library) in TEST() 233 FunctionLibraryDefinition function_library(OpRegistry::Global()); in TEST() local 247 SplitFunction(orig, {"t1"}, /*num_captured_inputs=*/0, function_library) in TEST() 262 FunctionLibraryDefinition function_library(OpRegistry::Global()); in TEST() local 278 SplitFunction(orig, {"i1"}, /*num_captured_inputs=*/0, function_library) in TEST() 296 FunctionLibraryDefinition function_library(OpRegistry::Global()); in TEST() local 313 SplitFunction(orig, {"add"}, num_captured, function_library) in TEST() 330 FunctionLibraryDefinition function_library(OpRegistry::Global()); in TEST() local 351 function_library) in TEST() [all …]
|
H A D | map_fusion.cc | 92 FunctionLibraryDefinition function_library(OpRegistry::Global(), in OptimizeAndCollectStats() local 102 auto make_fused_function = [&function_library, &output]( in OptimizeAndCollectStats() 107 function_library.Find(parent_fun.func().name()); in OptimizeAndCollectStats() 109 const FunctionDef* func = function_library.Find(fun.func().name()); in OptimizeAndCollectStats() 141 TF_RETURN_IF_ERROR(function_library.AddFunctionDef(*fused_function)); in OptimizeAndCollectStats()
|
H A D | filter_fusion.cc | 73 FunctionLibraryDefinition function_library(OpRegistry::Global(), in OptimizeAndCollectStats() local 87 function_library.Find(parent_fun.func().name()); in OptimizeAndCollectStats() 89 const FunctionDef* second_func = function_library.Find(fun.func().name()); in OptimizeAndCollectStats() 121 TF_RETURN_IF_ERROR(function_library.AddFunctionDef(*fused_predicate)); in OptimizeAndCollectStats()
|
H A D | map_and_filter_fusion.cc | 173 FunctionLibraryDefinition function_library(OpRegistry::Global(), in OptimizeAndCollectStats() local 190 auto make_fused_function = [&function_library, &output]( in OptimizeAndCollectStats() 195 function_library.Find(parent_fun.func().name()); in OptimizeAndCollectStats() 197 const FunctionDef* filter_func = function_library.Find(fun.func().name()); in OptimizeAndCollectStats() 235 TF_RETURN_IF_ERROR(function_library.AddFunctionDef(*fused_function)); in OptimizeAndCollectStats()
|
H A D | filter_parallelization.cc | 74 FunctionLibraryDefinition function_library(OpRegistry::Global(), in OptimizeAndCollectStats() local 85 auto* function = function_library.Find( in OptimizeAndCollectStats() 87 if (function_utils::IsFunctionStateful(function_library, *function, true)) { in OptimizeAndCollectStats()
|
H A D | map_parallelization.cc | 75 FunctionLibraryDefinition function_library(OpRegistry::Global(), in OptimizeAndCollectStats() local 87 function_library.Find(map_node->attr().at("f").func().name()); in OptimizeAndCollectStats() 88 if (function_utils::IsFunctionStateful(function_library, *function, true)) in OptimizeAndCollectStats()
|
H A D | noop_elimination.cc | 120 FunctionLibraryDefinition function_library(OpRegistry::Global(), in IsMapIdentity() local 123 function_library.Find(map_node.attr().at("f").func().name()); in IsMapIdentity() 126 if (function_utils::IsFunctionStateful(function_library, *fdef)) return false; in IsMapIdentity()
|
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/ |
H A D | graph_runner.cc | 103 Status GraphRunner::Run(Graph* graph, FunctionLibraryRuntime* function_library, in Run() argument 111 if (function_library && function_library->device() && in Run() 112 function_library->device()->device_type() != device_->device_type()) { in Run() 119 << function_library->device()->device_type() << " device."; in Run() 120 function_library = nullptr; in Run() 160 params.function_library = function_library; in Run() 162 params.create_kernel = [this, function_library, producer]( in Run() 165 return CreateNonCachedKernel(device_, function_library, props, producer, in Run()
|
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/ |
H A D | mlir_bridge_pass.cc | 169 const FunctionLibraryDefinition& function_library) const { in GetPassState() 185 graph, &function_library, config_proto, in GetPassState() 218 const FunctionLibraryDefinition& function_library) { in Run() argument 243 function_library); in Run() 268 const FunctionLibraryDefinition& function_library) const { in GetPassState() 282 graph, /*function_library=*/&function_library, config_proto, in GetPassState()
|
H A D | mlir_bridge_pass.h | 36 const FunctionLibraryDefinition& function_library) const override; 42 const FunctionLibraryDefinition& function_library) override; 55 const FunctionLibraryDefinition& function_library) const override;
|
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
H A D | functional_ops.cc | 50 return ctx->function_library()->Instantiate( in Instantiate() 127 auto lib = ctx->function_library(); in IfOp() 163 lib_(CHECK_NOTNULL(ctx_->function_library())) { in State() 213 auto lib = ctx->function_library(); in GetHandles() 244 auto lib = ctx->function_library(); in CaseOp() 252 auto lib = ctx->function_library(); in ComputeAsync() 289 lib_(CHECK_NOTNULL(ctx_->function_library())) { in State() 510 lib_(CHECK_NOTNULL(ctx_->function_library())) { in State() 606 auto lib = ctx->function_library(); in DoComputeSync() 666 auto lib = ctx->function_library(); in GetHandles() [all …]
|
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/ |
H A D | mlir_graph_optimization_pass_test.cc | 39 const FunctionLibraryDefinition& function_library)); 42 const FunctionLibraryDefinition& function_library)); 54 const FunctionLibraryDefinition& function_library)); 69 const FunctionLibraryDefinition& function_library)); 75 const FunctionLibraryDefinition& function_library) override { in Run() argument
|
H A D | mlir_graph_optimization_pass.h | 65 const FunctionLibraryDefinition& function_library) const = 0; 69 const FunctionLibraryDefinition& function_library) = 0; 148 const FunctionLibraryDefinition& function_library) const = 0;
|
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tfr/integration/ |
H A D | graph_decompose_pass.cc | 36 const FunctionLibraryDefinition& function_library) const { in GetPassState() 44 const FunctionLibraryDefinition& function_library) { in Run() argument 46 function_library) == MlirOptimizationPassState::Disabled) { in Run()
|
H A D | graph_decompose_pass.h | 39 const FunctionLibraryDefinition& function_library) const override; 45 const FunctionLibraryDefinition& function_library) override;
|
/aosp_15_r20/external/tensorflow/tensorflow/compiler/jit/ |
H A D | xla_compile_on_demand_op.cc | 117 &ctx->op_kernel(), &constant_input_indices, ctx->function_library())); in Compile() 133 return BuildXlaCompilationCache(ctx->device(), ctx->function_library(), in Compile() 138 **cache, *ctx->function_library(), ctx->device(), in Compile() 176 OP_REQUIRES(ctx, ctx->function_library(), in Compute()
|
H A D | xla_platform_info.cc | 169 const FunctionLibraryRuntime& function_library, DeviceBase* device, in GenerateCompilerOptions() argument 178 options.flib_def = function_library.GetFunctionLibraryDefinition(); in GenerateCompilerOptions() 179 options.graph_def_version = function_library.graph_def_version(); in GenerateCompilerOptions()
|
/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 1445 function_library); in TEST_F() 1485 std::vector<FunctionDef> function_library = {x_times_two}; in TEST_F() local 1499 function_library); in TEST_F() 1553 std::vector<FunctionDef> function_library = {mul_func}; in TEST_F() local 1567 function_library); in TEST_F() 1617 std::vector<FunctionDef> function_library = {mul_func}; in TEST_F() local 1636 function_library); in TEST_F() 1695 std::vector<FunctionDef> function_library = {mul_func}; in TEST_F() local 1732 function_library); in TEST_F() [all …]
|
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 396 ? ctx.function_library().Find(func_attr->func().name()) in FindFunctionCall() 401 return ctx.function_library().Find(node.op()); in FindFunctionCall() 717 const auto& flib = ctx->function_library(); in SpecializeFunction() 764 TF_RETURN_IF_ERROR(ctx->function_library().AddFunctionDef(specialized_func)); in SpecializeFunction() 1482 const string grad_func = ctx.function_library().FindGradient(func_name); in RunFunctionOptimizerPass() 1510 PruneFunctionLibrary(ctx.function_library(), *optimized_graph); in RunFunctionOptimizerPass()
|
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/costs/ |
H A D | op_context.h | 32 const FunctionDefLibrary* function_library; // Not owned. member 38 OpContext() { function_library = nullptr; } in OpContext()
|
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/verifiers/ |
H A D | structure_verifier.cc | 37 FunctionLibraryDefinition function_library(OpRegistry::Global(), in Verify() local 40 graph, function_library)); in Verify()
|
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tf2xla/ |
H A D | mlir_bridge_rollout_policy.cc | 24 const FunctionLibraryDefinition* function_library, in GetMlirBridgeRolloutPolicy() argument 41 const FunctionLibraryDefinition* function_library, in LogGraphFeatures() argument
|
H A D | mlir_bridge_rollout_policy.h | 57 const FunctionLibraryDefinition* function_library, 74 const FunctionLibraryDefinition* function_library,
|
/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() argument 44 const tensorflow::FunctionLibraryDefinition& function_library) override;
|
H A D | graph_optimization_pass.cc | 37 const tensorflow::FunctionLibraryDefinition& function_library) { in Run() argument 39 function_library) == in Run()
|