Home
last modified time | relevance | path

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

123

/external/tensorflow/tensorflow/core/common_runtime/
Dgraph_runner.cc101 Status GraphRunner::Run(Graph* graph, FunctionLibraryRuntime* function_library, in Run() argument
109 if (function_library && function_library->device() && in Run()
110 function_library->device()->device_type() != device_->device_type()) { in Run()
117 << function_library->device()->device_type() << " device."; in Run()
118 function_library = nullptr; in Run()
159 params.function_library = function_library; in Run()
161 params.create_kernel = [this, function_library, producer](const NodeDef& ndef, in Run()
163 return CreateNonCachedKernel(device_, function_library, ndef, producer, in Run()
Dconstant_folding.h64 FunctionLibraryRuntime* function_library, Env* env,
Dgraph_runner.h58 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 get_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()
142 TF_RETURN_IF_ERROR(function_library.AddFunctionDef(*fused_function)); in OptimizeAndCollectStats()
Dmap_and_filter_fusion.cc108 FunctionLibraryDefinition function_library(OpRegistry::Global(), in OptimizeAndCollectStats() local
122 auto make_fused_function = [&function_library, &output]( in OptimizeAndCollectStats()
127 function_library.Find(parent_fun.func().name()); in OptimizeAndCollectStats()
129 const FunctionDef* filter_func = function_library.Find(fun.func().name()); in OptimizeAndCollectStats()
163 TF_RETURN_IF_ERROR(function_library.AddFunctionDef(*fused_function)); in OptimizeAndCollectStats()
Dfilter_fusion.cc71 FunctionLibraryDefinition function_library(OpRegistry::Global(), in OptimizeAndCollectStats() local
84 function_library.Find(parent_fun.func().name()); in OptimizeAndCollectStats()
86 const FunctionDef* second_func = function_library.Find(fun.func().name()); in OptimizeAndCollectStats()
121 TF_RETURN_IF_ERROR(function_library.AddFunctionDef(*fused_predicate)); in OptimizeAndCollectStats()
Dmap_parallelization.cc62 FunctionLibraryDefinition function_library(OpRegistry::Global(), in OptimizeAndCollectStats() local
74 function_library.Find(map_node->attr().at("f").func().name()); in OptimizeAndCollectStats()
75 if (function_utils::IsFunctionStateful(function_library, *function, true)) in OptimizeAndCollectStats()
Dhoist_random_uniform.cc232 FunctionLibraryDefinition function_library(OpRegistry::Global(), in OptimizeAndCollectStats() local
247 const FunctionDef* func = function_library.Find(fun.func().name()); in OptimizeAndCollectStats()
251 if (!CanHoistRandomUniform(*func, function_library, in OptimizeAndCollectStats()
Dmap_vectorization.cc447 const FunctionLibraryDefinition& function_library, in FindMapAndBatchPattern() argument
497 map_fn = function_library.Find(map_node->attr().at("f").func().name()); in FindMapAndBatchPattern()
499 if (function_utils::IsFunctionStateful(function_library, *map_fn)) { in FindMapAndBatchPattern()
521 FunctionLibraryDefinition function_library(OpRegistry::Global(), *library); in OptimizeAndCollectStats() local
527 if (!FindMapAndBatchPattern(graph, node, function_library, &batch_node, in OptimizeAndCollectStats()
/external/tensorflow/tensorflow/core/kernels/
Dfunctional_ops.cc130 auto lib = ctx->function_library(); in IfOp()
139 auto lib = ctx->function_library(); in ComputeAsync()
174 lib_(CHECK_NOTNULL(ctx_->function_library())) { in State()
218 auto lib = ctx->function_library(); in CaseOp()
226 auto lib = ctx->function_library(); in ComputeAsync()
263 lib_(CHECK_NOTNULL(ctx_->function_library())) { in State()
333 auto lib = ctx->function_library(); in ComputeAsync()
364 lib_(CHECK_NOTNULL(ctx_->function_library())) { in State()
501 auto lib = ctx->function_library(); in ForOp()
523 lib_(CHECK_NOTNULL(ctx_->function_library())), in State()
/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/core/grappler/optimizers/
Dfunction_optimizer.cc288 const FunctionLibraryDefinition& function_library() const { in function_library() function in tensorflow::grappler::__anon35816ea30111::FunctionOptimizerContext
481 ? ctx.function_library().Find(func_attr->func().name()) in FindFunctionCall()
486 return ctx.function_library().Find(node.op()); in FindFunctionCall()
804 const auto& flib = ctx->function_library(); in SpecializeFunction()
939 return !IsFreeOfSideEffect(node, &ctx.function_library()); in IsInlinableDirectFunctionCall()
1022 ctx.function_library(), in InlineDirectFunctionCall()
1155 Graph graph(ctx->function_library()); in InlineSymbolicGradient()
1325 const FunctionDef* func = ctx.function_library().Find(node.op()); in MaybeDeadOutputs()
1330 ctx.function_library(), ctx.graph_version(), &func_item)); in MaybeDeadOutputs()
1426 IsStateful(func_body_node, &ctx.function_library()); in CheckThatSideEffectsWillExecute()
[all …]
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_api_info.cc157 const FunctionDefLibrary& function_library) { in Init() argument
161 for (const auto& function : function_library.function()) { in Init()
Dfunction_api_info.h77 Status Init(const FunctionDefLibrary& function_library);
/external/tensorflow/tensorflow/core/kernels/data/
Ddataset_utils.cc30 TF_RETURN_IF_ERROR(ctx->function_library()->Instantiate( in ComputeShortCircuitIndices()
33 Status s = ctx->function_library()->ReleaseHandle(fn_handle); in ComputeShortCircuitIndices()
41 if (ctx->function_library()->IsStateful(func.name())) { in ComputeShortCircuitIndices()
47 ctx->function_library()->GetFunctionBody(fn_handle); in ComputeShortCircuitIndices()
Dgraph_rewrite_dataset.cc38 params.flib_def = ctx->function_library()->GetFunctionLibraryDefinition(); in Optimize()
55 TF_RETURN_IF_ERROR(ctx->function_library()->Clone(&flib_def_, &pflr_, &lib_)); in Optimize()
69 GraphRunner graph_runner(ctx->function_library()->device()); in Optimize()
Diterator_ops.cc150 TF_RETURN_IF_ERROR(ctx->function_library()->Clone(&flib_def, &pflr, &lib)); in Restore()
217 *ctx->function_library()->GetFunctionLibraryDefinition()); in SetIteratorFromDataset()
229 ctx->function_library()->Clone(&flib_def, &pflr, &lib)); in SetIteratorFromDataset()
341 params.flib_def = ctx->function_library()->GetFunctionLibraryDefinition(); in InitializeFromIterator()
432 OP_REQUIRES_OK(context, context->function_library()->Clone( in Compute()
488 *ctx->function_library()->GetFunctionLibraryDefinition()); in CreatePrivateFLR()
514 context->function_library()->Clone(&flib_def, &pflr, &lib)); in Compute()
843 TF_RETURN_IF_ERROR(ctx->function_library()->Clone(&flib_def, &pflr, &lib)); in TryInit()
868 TF_RETURN_IF_ERROR(ctx->function_library()->Instantiate( in TryInit()
887 ctx->function_library()->Run(opts, f_handle, {}, &return_values, in TryInit()
Ddataset_ops.cc37 params.flib_def = ctx->function_library()->GetFunctionLibraryDefinition(); in Compute()
Dsingle_threaded_executor.cc217 params.function_library = params_.function_library; in RunAsync()
/external/tensorflow/tensorflow/core/framework/
Ddataset.h285 function_library(ctx->function_library()), in Params()
297 lib(ctx->function_library()), in Params()
301 reinterpret_cast<DeviceBase*>(ctx->function_library()->device()); in Params()
321 std::shared_ptr<const FunctionLibraryDefinition> function_library = nullptr; member
366 std::shared_ptr<const FunctionLibraryDefinition> function_library() { in function_library() function
367 return params_.function_library; in function_library()
/external/tensorflow/tensorflow/compiler/jit/
Dxla_fusion_optimizer.cc136 FunctionLibraryDefinition function_library(OpRegistry::Global(), in Optimize() local
138 Graph graph(function_library); in Optimize()
/external/tensorflow/tensorflow/compiler/tf2xla/
Dxla_op_kernel.h230 FunctionLibraryRuntime* function_library() const { in function_library() function
231 return context_->function_library(); in function_library()
Dgraph_compiler.cc63 /*compile_time_const_nodes=*/nullptr, ctx->function_library())); in PrepareArguments()
311 params->function_library = flib_; in PartiallySetupParams()

123