Searched refs:GraphCollector (Results 1 – 8 of 8) 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()
81 GraphCollector* graph_collector) { in Init()
154 GraphCollector* graph_collector) { in Run()
165 GraphCollector* graph_collector) { in Run()
210 GraphCollector* graph_collector) { in Run()
308 GraphCollector* graph_collector) { in Run()
|
| D | execute_node.h | 37 StepStats* maybe_step_stats, GraphCollector* graph_collector, in ExecuteNode()
85 tensorflow::GraphCollector* graph_collector_;
|
| D | execute.h | 50 GraphCollector* graph_collector,
|
| D | context.h | 145 GraphCollector* GetGraphCollector() { return &graph_collector_; } in GetGraphCollector()
293 GraphCollector graph_collector_;
|
| D | execute.cc | 428 GraphCollector* graph_collector = nullptr; in EagerLocalExecute()
481 GraphCollector* graph_collector = nullptr; in EagerLocalExecute()
916 GraphCollector* graph_collector, in EagerKernelExecute()
954 GraphCollector* collector = ctx->GetGraphCollector(); in EagerKernelExecute()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | op_kernel.h | 526 struct GraphCollector { struct
534 GraphCollector() : dirty(false) {} in GraphCollector() argument
670 GraphCollector* graph_collector = nullptr;
798 GraphCollector* graph_collector() { return params_->graph_collector; } in graph_collector()
|
| D | function.h | 583 GraphCollector* graph_collector = nullptr;
|