Home
last modified time | relevance | path

Searched refs:function_library (Results 1 – 25 of 87) sorted by relevance

1234

/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/data/
H A Dsplit_utils_test.cc188 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 Dmap_fusion.cc92 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 Dfilter_fusion.cc73 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 Dmap_and_filter_fusion.cc173 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 Dfilter_parallelization.cc74 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 Dmap_parallelization.cc75 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 Dnoop_elimination.cc120 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 Dgraph_runner.cc103 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 Dmlir_bridge_pass.cc169 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 Dmlir_bridge_pass.h36 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 Dfunctional_ops.cc50 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 Dmlir_graph_optimization_pass_test.cc39 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 Dmlir_graph_optimization_pass.h65 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 Dgraph_decompose_pass.cc36 const FunctionLibraryDefinition& function_library) const { in GetPassState()
44 const FunctionLibraryDefinition& function_library) { in Run() argument
46 function_library) == MlirOptimizationPassState::Disabled) { in Run()
H A Dgraph_decompose_pass.h39 const FunctionLibraryDefinition& function_library) const override;
45 const FunctionLibraryDefinition& function_library) override;
/aosp_15_r20/external/tensorflow/tensorflow/compiler/jit/
H A Dxla_compile_on_demand_op.cc117 &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 Dxla_platform_info.cc169 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 Dfunction_optimizer_test.cc1436 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 Dfunction_optimizer.cc274 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 Dop_context.h32 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 Dstructure_verifier.cc37 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 Dmlir_bridge_rollout_policy.cc24 const FunctionLibraryDefinition* function_library, in GetMlirBridgeRolloutPolicy() argument
41 const FunctionLibraryDefinition* function_library, in LogGraphFeatures() argument
H A Dmlir_bridge_rollout_policy.h57 const FunctionLibraryDefinition* function_library,
74 const FunctionLibraryDefinition* function_library,
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
H A Dgraph_optimization_pass.h34 const tensorflow::FunctionLibraryDefinition& function_library) in GetPassState() argument
44 const tensorflow::FunctionLibraryDefinition& function_library) override;
H A Dgraph_optimization_pass.cc37 const tensorflow::FunctionLibraryDefinition& function_library) { in Run() argument
39 function_library) == in Run()

1234