/external/tensorflow/tensorflow/core/common_runtime/ |
D | graph_runner.cc | 101 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()
|
D | constant_folding.h | 64 FunctionLibraryRuntime* function_library, Env* env,
|
D | graph_runner.h | 58 Status Run(Graph* graph, FunctionLibraryRuntime* function_library,
|
/external/tensorflow/tensorflow/core/grappler/optimizers/data/ |
D | map_fusion.cc | 91 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()
|
D | map_and_filter_fusion.cc | 108 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()
|
D | filter_fusion.cc | 71 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()
|
D | map_parallelization.cc | 62 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()
|
D | hoist_random_uniform.cc | 232 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()
|
D | map_vectorization.cc | 447 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/ |
D | functional_ops.cc | 130 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/ |
D | op_context.h | 31 const FunctionDefLibrary* function_library; // Not owned. member 33 OpContext() { function_library = nullptr; } in OpContext()
|
/external/tensorflow/tensorflow/core/grappler/verifiers/ |
D | structure_verifier.cc | 37 FunctionLibraryDefinition function_library(OpRegistry::Global(), in Verify() local 40 graph, function_library)); in Verify()
|
/external/tensorflow/tensorflow/core/grappler/optimizers/ |
D | function_optimizer.cc | 288 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 …]
|
D | function_optimizer_test.cc | 1437 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 …]
|
D | function_api_info.cc | 157 const FunctionDefLibrary& function_library) { in Init() argument 161 for (const auto& function : function_library.function()) { in Init()
|
D | function_api_info.h | 77 Status Init(const FunctionDefLibrary& function_library);
|
/external/tensorflow/tensorflow/core/kernels/data/ |
D | dataset_utils.cc | 30 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()
|
D | graph_rewrite_dataset.cc | 38 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()
|
D | iterator_ops.cc | 150 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()
|
D | dataset_ops.cc | 37 params.flib_def = ctx->function_library()->GetFunctionLibraryDefinition(); in Compute()
|
D | single_threaded_executor.cc | 217 params.function_library = params_.function_library; in RunAsync()
|
/external/tensorflow/tensorflow/core/framework/ |
D | dataset.h | 285 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/ |
D | xla_fusion_optimizer.cc | 136 FunctionLibraryDefinition function_library(OpRegistry::Global(), in Optimize() local 138 Graph graph(function_library); in Optimize()
|
/external/tensorflow/tensorflow/compiler/tf2xla/ |
D | xla_op_kernel.h | 230 FunctionLibraryRuntime* function_library() const { in function_library() function 231 return context_->function_library(); in function_library()
|
D | graph_compiler.cc | 63 /*compile_time_const_nodes=*/nullptr, ctx->function_library())); in PrepareArguments() 311 params->function_library = flib_; in PartiallySetupParams()
|