Home
last modified time | relevance | path

Searched refs:gb_per_sec (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/core/grappler/costs/
Dcost_estimator.h40 double gb_per_sec; member
50 gb_per_sec(INFINITY), in DeviceInfo()
56 gb_per_sec(input.gb_per_sec), in DeviceInfo()
60 DeviceInfo(double gigaops, double gb_per_sec,
64 gb_per_sec(gb_per_sec), in gigaops()
Dop_level_cost_estimator.cc722 double gb_per_sec = -1; in GetDeviceInfo() local
729 if (gb_per_sec < 0) { in GetDeviceInfo()
731 gb_per_sec = device.bandwidth() / 1e6; in GetDeviceInfo()
733 gb_per_sec = 32; in GetDeviceInfo()
756 gb_per_sec = device.bandwidth() / 1e6; in GetDeviceInfo()
758 gb_per_sec = 100; in GetDeviceInfo()
764 gb_per_sec = 12; // default PCIe x16 gen3. in GetDeviceInfo()
767 << " gb_per_sec: " << gb_per_sec; in GetDeviceInfo()
769 return DeviceInfo(gflops, gb_per_sec); in GetDeviceInfo()
832 if (device_info.gigaops <= 0 || device_info.gb_per_sec <= 0 || in PredictOpCountBasedCost()
[all …]