Searched refs:GraphCollector (Results 1 – 8 of 8) sorted by relevance
98 GraphCollector* graph_collector, in ExecuteNode()142 GraphCollector* graph_collector_;154 GraphCollector* graph_collector, in AsyncExecuteNode()237 GraphCollector* graph_collector_;
106 GraphCollector* graph_collector) = 0;206 GraphCollector* graph_collector) override;292 GraphCollector* graph_collector);295 GraphCollector* graph_collector) override;
54 GraphCollector* graph_collector, CancellationManager* cancellation_manager,
102 GraphCollector* graph_collector) { in Init()135 GraphCollector* graph_collector) { in InstantiateFunc()225 GraphCollector* graph_collector) { in Init()
508 GraphCollector* graph_collector = nullptr; in GetOrCreateKernelAndDevice()613 GraphCollector* graph_collector = nullptr; in AddOrExecuteNode()1049 GraphCollector* collector = ctx->GetGraphCollector(); in CollectGraphs()1111 GraphCollector* graph_collector, CancellationManager* cancellation_manager, in EagerKernelExecute()1318 GraphCollector* graph_collector, CancellationManager* cancellation_manager, in EagerKernelExecuteAsync()1398 GraphCollector* graph_collector = nullptr; in EagerLocalExecuteAsync()
210 GraphCollector* GetGraphCollector() { return &graph_collector_; } in GetGraphCollector()634 GraphCollector graph_collector_;
544 struct GraphCollector { struct552 GraphCollector() : dirty(false) {} in GraphCollector() argument688 GraphCollector* graph_collector = nullptr;829 GraphCollector* graph_collector() { return params_->graph_collector; } in graph_collector()
729 GraphCollector* graph_collector = nullptr;