Home
last modified time | relevance | path

Searched refs:graph_collector (Results 1 – 10 of 10) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/eager/
Dexecute_node.h98 GraphCollector* graph_collector, in ExecuteNode() argument
107 graph_collector_(graph_collector), in ExecuteNode()
154 GraphCollector* graph_collector, in AsyncExecuteNode() argument
163 graph_collector_(graph_collector), in AsyncExecuteNode()
Dkernel_and_device.cc102 GraphCollector* graph_collector) { in Init() argument
135 GraphCollector* graph_collector) { in InstantiateFunc() argument
204 options.graph_collector = graph_collector; in InstantiateFunc()
225 GraphCollector* graph_collector) { in Init() argument
227 InstantiateFunc(log_device_placement, ndef, graph_collector)); in Init()
Dexecute.cc508 GraphCollector* graph_collector = nullptr; in GetOrCreateKernelAndDevice() local
510 graph_collector = ctx.GetGraphCollector(); in GetOrCreateKernelAndDevice()
542 kernel->Init(ctx.LogDevicePlacement(), ndef, graph_collector)); in GetOrCreateKernelAndDevice()
613 GraphCollector* graph_collector = nullptr; in AddOrExecuteNode() local
615 graph_collector = ctx.GetGraphCollector(); in AddOrExecuteNode()
649 &ctx, *inputs, remote_func_params, std::move(kernel), graph_collector, in AddOrExecuteNode()
667 ExecuteNode node(&ctx, *inputs, remote_func_params, kernel, graph_collector, in AddOrExecuteNode()
1111 GraphCollector* graph_collector, CancellationManager* cancellation_manager, in EagerKernelExecute() argument
1130 if (graph_collector != nullptr) { in EagerKernelExecute()
1318 GraphCollector* graph_collector, CancellationManager* cancellation_manager, in EagerKernelExecuteAsync() argument
[all …]
Dkernel_and_device.h106 GraphCollector* graph_collector) = 0;
206 GraphCollector* graph_collector) override;
292 GraphCollector* graph_collector);
295 GraphCollector* graph_collector) override;
Dexecute.h54 GraphCollector* graph_collector, CancellationManager* cancellation_manager,
/external/tensorflow/tensorflow/core/kernels/
Dpartitioned_function_ops.cc205 opts.graph_collector = ctx->graph_collector(); in Instantiate()
/external/tensorflow/tensorflow/core/common_runtime/
Dprocess_function_library_runtime.cc721 if (options.graph_collector != nullptr) { in InstantiateMultiDevice()
725 options.graph_collector->CollectRawGraph(def); in InstantiateMultiDevice()
846 if (options.graph_collector != nullptr) { in InstantiateMultiDevice()
850 options.graph_collector->CollectOptimizedGraph(def); in InstantiateMultiDevice()
889 if (options.graph_collector != nullptr) { in InstantiateMultiDevice()
894 options.graph_collector->CollectPartitionedGraph(def); in InstantiateMultiDevice()
/external/tensorflow/tensorflow/core/framework/
Dop_kernel.h688 GraphCollector* graph_collector = nullptr; member
829 GraphCollector* graph_collector() { return params_->graph_collector; } in graph_collector() function
Dfunction.h729 GraphCollector* graph_collector = nullptr; member
/external/tensorflow/tensorflow/core/common_runtime/device/
Ddevice_event_mgr_test.cc294 params->graph_collector = nullptr; in PrepOpKernel()