Searched refs:set_core_count (Results 1 – 5 of 5) sorted by relevance
312 void set_core_count(int value) { device_description_->core_count_ = value; } in set_core_count() function
367 builder.set_core_count(description->core_count); in CreateDeviceDescription()
911 builder.set_core_count( in CreateDeviceDescription()
1072 builder.set_core_count( in CreateDeviceDescription()
329 pod_info_.set_core_count(pod_info_.local_core_size()); in PodTpuDriver()