Searched refs:device_context_map (Results 1 – 8 of 8) sorted by relevance
78 DeviceContextMap* device_context_map) { in FillContextMap() argument81 device_context_map->resize(graph->num_node_ids()); in FillContextMap()84 (*device_context_map)[n->id()] = device_context_; in FillContextMap()
220 DeviceContextMap* device_context_map) override;
120 DeviceContextMap* device_context_map) override { in FillContextMap() argument121 return underlying_->FillContextMap(graph, device_context_map); in FillContextMap()
166 DeviceContextMap* device_context_map) { in FillContextMap() argument
358 DeviceContextMap* device_context_map) { in FillContextMap() argument364 device_context_map->resize(graph->num_node_ids()); in FillContextMap()368 (*device_context_map)[n->id()] = device_context; in FillContextMap()
148 DeviceContextMap* device_context_map) override
80 DeviceContextMap* device_context_map) override;
455 DeviceContextMap* device_context_map) { in FillContextMap() argument474 device_context_map->resize(graph->num_node_ids()); in FillContextMap()483 (*device_context_map)[n->id()] = ctx; in FillContextMap()