Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/common_runtime/eager/
Dkernel_and_device.h59 virtual Status Init(const NodeDef& ndef, GraphCollector* graph_collector) = 0;
84 GraphCollector* graph_collector) = 0;
90 GraphCollector* graph_collector) = 0;
145 Status Init(const NodeDef& ndef, GraphCollector* graph_collector) override;
149 StepStats* step_stats, GraphCollector* graph_collector) override;
154 StepStats* step_stats, GraphCollector* graph_collector) override;
207 Status Init(const NodeDef& ndef, GraphCollector* graph_collector) override;
211 StepStats* step_stats, GraphCollector* graph_collector) override;
215 StepStats* step_stats, GraphCollector* graph_collector) override;
Dkernel_and_device.cc68 GraphCollector* graph_collector) { in Init() argument
81 GraphCollector* graph_collector) { in Init() argument
143 options.graph_collector = graph_collector; in Init()
154 GraphCollector* graph_collector) { in Run() argument
159 graph_collector); in Run()
165 GraphCollector* graph_collector) { in Run() argument
173 graph_collector); in Run()
210 GraphCollector* graph_collector) { in Run() argument
261 params.graph_collector = graph_collector; in Run()
308 GraphCollector* graph_collector) { in Run() argument
Dexecute_node.h37 StepStats* maybe_step_stats, GraphCollector* graph_collector, in ExecuteNode() argument
46 graph_collector_(graph_collector), in ExecuteNode()
Dexecute.cc428 GraphCollector* graph_collector = nullptr; in EagerLocalExecute() local
430 graph_collector = ctx->GetGraphCollector(); in EagerLocalExecute()
456 status = kernel->Init(ndef, graph_collector); in EagerLocalExecute()
481 GraphCollector* graph_collector = nullptr; in EagerLocalExecute() local
483 graph_collector = ctx->GetGraphCollector(); in EagerLocalExecute()
514 graph_collector, output_dtypes, *retvals); in EagerLocalExecute()
520 maybe_step_stats, graph_collector, in EagerLocalExecute()
916 GraphCollector* graph_collector, in EagerKernelExecute() argument
945 maybe_step_stats, graph_collector)); in EagerKernelExecute()
949 graph_collector)); in EagerKernelExecute()
[all …]
Dexecute.h50 GraphCollector* graph_collector,
/external/tensorflow/tensorflow/core/common_runtime/
Dprocess_function_library_runtime.cc561 if (options.graph_collector != nullptr) { in InstantiateMultiDevice()
565 options.graph_collector->CollectRawGraph(def); in InstantiateMultiDevice()
642 if (options.graph_collector != nullptr) { in InstantiateMultiDevice()
646 options.graph_collector->CollectOptimizedGraph(def); in InstantiateMultiDevice()
653 if (options.graph_collector != nullptr) { in InstantiateMultiDevice()
658 options.graph_collector->CollectPartitionedGraph(def); in InstantiateMultiDevice()
/external/tensorflow/tensorflow/core/kernels/
Dpartitioned_function_ops.cc180 opts.graph_collector = ctx->graph_collector(); in Instantiate()
/external/tensorflow/tensorflow/core/framework/
Dop_kernel.h670 GraphCollector* graph_collector = nullptr; member
798 GraphCollector* graph_collector() { return params_->graph_collector; } in graph_collector() function
Dfunction.h583 GraphCollector* graph_collector = nullptr; member