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.h38 double gb_per_sec; member
48 gb_per_sec(INFINITY), in DeviceInfo()
54 gb_per_sec(input.gb_per_sec), in DeviceInfo()
58 DeviceInfo(double gigaops, double gb_per_sec,
62 gb_per_sec(gb_per_sec), in gigaops()
Dop_level_cost_estimator.cc400 double gb_per_sec = -1; in GetDeviceInfo() local
407 if (gb_per_sec < 0) { in GetDeviceInfo()
409 gb_per_sec = device.bandwidth() / 1e6; in GetDeviceInfo()
411 gb_per_sec = 32; in GetDeviceInfo()
434 gb_per_sec = device.bandwidth() / 1e6; in GetDeviceInfo()
436 gb_per_sec = 100; in GetDeviceInfo()
440 << " gb_per_sec: " << gb_per_sec; in GetDeviceInfo()
443 DCHECK_LT(0, gb_per_sec) << device.DebugString(); in GetDeviceInfo()
445 return DeviceInfo(gflops, gb_per_sec); in GetDeviceInfo()
515 if (device_info.gigaops <= 0 || device_info.gb_per_sec <= 0 || in PredictOpCountBasedCost()
[all …]