Searched refs:graph_collector (Results 1 – 9 of 9) sorted by relevance
/external/tensorflow/tensorflow/core/common_runtime/eager/ |
D | kernel_and_device.h | 59 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;
|
D | kernel_and_device.cc | 68 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
|
D | execute_node.h | 37 StepStats* maybe_step_stats, GraphCollector* graph_collector, in ExecuteNode() argument 46 graph_collector_(graph_collector), in ExecuteNode()
|
D | execute.cc | 428 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 …]
|
D | execute.h | 50 GraphCollector* graph_collector,
|
/external/tensorflow/tensorflow/core/common_runtime/ |
D | process_function_library_runtime.cc | 561 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/ |
D | partitioned_function_ops.cc | 180 opts.graph_collector = ctx->graph_collector(); in Instantiate()
|
/external/tensorflow/tensorflow/core/framework/ |
D | op_kernel.h | 670 GraphCollector* graph_collector = nullptr; member 798 GraphCollector* graph_collector() { return params_->graph_collector; } in graph_collector() function
|
D | function.h | 583 GraphCollector* graph_collector = nullptr; member
|