Home
last modified time | relevance | path

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

123

/external/tensorflow/tensorflow/core/common_runtime/
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()
Dlocal_executor_params.h41 FunctionLibraryRuntime* function_library = nullptr; member
Dconstant_folding.h64 FunctionLibraryRuntime* function_library, Env* env,
Dgraph_runner.h59 Status Run(Graph* graph, FunctionLibraryRuntime* function_library,
/external/tensorflow/tensorflow/core/grappler/optimizers/data/
Dmap_fusion.cc91 FunctionLibraryDefinition function_library(OpRegistry::Global(), in OptimizeAndCollectStats() local
101 auto make_fused_function = [&function_library, &output]( in OptimizeAndCollectStats()
106 function_library.Find(parent_fun.func().name()); in OptimizeAndCollectStats()
108 const FunctionDef* func = function_library.Find(fun.func().name()); in OptimizeAndCollectStats()
140 TF_RETURN_IF_ERROR(function_library.AddFunctionDef(*fused_function)); in OptimizeAndCollectStats()
Dfilter_fusion.cc71 FunctionLibraryDefinition function_library(OpRegistry::Global(), in OptimizeAndCollectStats() local
85 function_library.Find(parent_fun.func().name()); in OptimizeAndCollectStats()
87 const FunctionDef* second_func = function_library.Find(fun.func().name()); in OptimizeAndCollectStats()
119 TF_RETURN_IF_ERROR(function_library.AddFunctionDef(*fused_predicate)); in OptimizeAndCollectStats()
Dmap_parallelization.cc76 FunctionLibraryDefinition function_library(OpRegistry::Global(), in OptimizeAndCollectStats() local
88 function_library.Find(map_node->attr().at("f").func().name()); in OptimizeAndCollectStats()
89 if (function_utils::IsFunctionStateful(function_library, *function, true)) in OptimizeAndCollectStats()
Dmap_and_filter_fusion.cc177 FunctionLibraryDefinition function_library(OpRegistry::Global(), in OptimizeAndCollectStats() local
194 auto make_fused_function = [&function_library, &output]( in OptimizeAndCollectStats()
199 function_library.Find(parent_fun.func().name()); in OptimizeAndCollectStats()
201 const FunctionDef* filter_func = function_library.Find(fun.func().name()); in OptimizeAndCollectStats()
239 TF_RETURN_IF_ERROR(function_library.AddFunctionDef(*fused_function)); in OptimizeAndCollectStats()
Dnoop_elimination.cc110 FunctionLibraryDefinition function_library(OpRegistry::Global(), in IsMapIdentity() local
113 function_library.Find(map_node.attr().at("f").func().name()); in IsMapIdentity()
116 if (function_utils::IsFunctionStateful(function_library, *fdef)) return false; in IsMapIdentity()
Dhoist_random_uniform.cc242 FunctionLibraryDefinition function_library(OpRegistry::Global(), in OptimizeAndCollectStats() local
257 const FunctionDef* func = function_library.Find(fun.func().name()); in OptimizeAndCollectStats()
261 if (!CanHoistRandomUniform(*func, function_library, in OptimizeAndCollectStats()
Dmap_vectorization.cc468 const FunctionLibraryDefinition& function_library, in FindMapAndBatchPattern() argument
519 map_fn = function_library.Find(map_node->attr().at("f").func().name()); in FindMapAndBatchPattern()
521 if (function_utils::IsFunctionStateful(function_library, *map_fn)) { in FindMapAndBatchPattern()
543 FunctionLibraryDefinition function_library(OpRegistry::Global(), *library); in OptimizeAndCollectStats() local
549 if (!FindMapAndBatchPattern(graph, node, function_library, &batch_node, in OptimizeAndCollectStats()
/external/tensorflow/tensorflow/core/grappler/costs/
Dop_context.h31 const FunctionDefLibrary* function_library; // Not owned. member
33 OpContext() { function_library = nullptr; } in OpContext()
/external/tensorflow/tensorflow/core/grappler/verifiers/
Dstructure_verifier.cc37 FunctionLibraryDefinition function_library(OpRegistry::Global(), in Verify() local
40 graph, function_library)); in Verify()
/external/tensorflow/tensorflow/compiler/jit/
Dxla_platform_info.cc132 const FunctionLibraryRuntime& function_library, DeviceBase* device, in GenerateCompilerOptions() argument
141 options.flib_def = function_library.GetFunctionLibraryDefinition(); in GenerateCompilerOptions()
142 options.graph_def_version = function_library.graph_def_version(); in GenerateCompilerOptions()
Dxla_compile_on_demand_op.cc109 &ctx->op_kernel(), &constant_input_indices, ctx->function_library())); in Compile()
131 **cache, *ctx->function_library(), ctx->device(), in Compile()
170 OP_REQUIRES(ctx, ctx->function_library(), in Compute()
Dxla_platform_info.h106 const FunctionLibraryRuntime& function_library, DeviceBase* device,
/external/tensorflow/tensorflow/core/kernels/
Dfunctional_ops.cc52 return ctx->function_library()->Instantiate( in Instantiate()
129 auto lib = ctx->function_library(); in IfOp()
165 lib_(CHECK_NOTNULL(ctx_->function_library())) { in State()
215 auto lib = ctx->function_library(); in GetHandles()
246 auto lib = ctx->function_library(); in CaseOp()
254 auto lib = ctx->function_library(); in ComputeAsync()
291 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 …]
Dpartitioned_function_ops.cc87 FunctionLibraryRuntime* lib = ctx->function_library(); in ComputeAsync()
173 const auto* config = (ctx->function_library()) in Instantiate()
174 ? ctx->function_library()->config_proto() in Instantiate()
Dfunction_ops.cc214 FunctionLibraryRuntime* lib = ctx->function_library(); in ComputeAsync()
272 FunctionLibraryRuntime* lib = ctx->function_library(); in ComputeAsync()
306 const auto* config = (ctx->function_library()) in ComputeAsync()
307 ? ctx->function_library()->config_proto() in ComputeAsync()
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dfunction_optimizer_test.cc1437 std::vector<FunctionDef> function_library = {x_times_two}; in TEST_F() local
1446 function_library); in TEST_F()
1486 std::vector<FunctionDef> function_library = {x_times_two}; in TEST_F() local
1500 function_library); in TEST_F()
1554 std::vector<FunctionDef> function_library = {mul_func}; in TEST_F() local
1568 function_library); in TEST_F()
1618 std::vector<FunctionDef> function_library = {mul_func}; in TEST_F() local
1637 function_library); in TEST_F()
1696 std::vector<FunctionDef> function_library = {mul_func}; in TEST_F() local
1733 function_library); in TEST_F()
[all …]
Dfunction_optimizer.cc271 const FunctionLibraryDefinition& function_library() const { in function_library() function in tensorflow::grappler::__anon36252d010111::FunctionOptimizerContext
274 FunctionLibraryDefinition& function_library() { return function_library_; } in function_library() function in tensorflow::grappler::__anon36252d010111::FunctionOptimizerContext
393 ? ctx.function_library().Find(func_attr->func().name()) in FindFunctionCall()
398 return ctx.function_library().Find(node.op()); in FindFunctionCall()
714 const auto& flib = ctx->function_library(); in SpecializeFunction()
761 TF_RETURN_IF_ERROR(ctx->function_library().AddFunctionDef(specialized_func)); in SpecializeFunction()
1472 const string grad_func = ctx.function_library().FindGradient(func_name); in RunFunctionOptimizerPass()
1500 PruneFunctionLibrary(ctx.function_library(), *optimized_graph); in RunFunctionOptimizerPass()
Dfunction_api_info.h77 Status Init(const FunctionDefLibrary& function_library);
Dfunction_api_info.cc157 const FunctionDefLibrary& function_library) { in Init() argument
161 for (const auto& function : function_library.function()) { in Init()
/external/tensorflow/tensorflow/compiler/tf2xla/
Dxla_op_kernel.h258 FunctionLibraryRuntime* function_library() const { in function_library() function
259 return context_->function_library(); in function_library()
/external/llvm-project/mlir/test/Analysis/
Dtest-shape-fn-report.mlir14 shape.function_library @shape_lib {

123