Searched refs:hardware_type (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/core/profiler/convert/ |
| D | op_stats_combiner.h | 31 bool IsCoordinator(bool no_accelerator_in_system, HardwareType hardware_type);
56 OpStatsInfo(const OpStats* op_stats, HardwareType hardware_type, in OpStatsInfo()
59 hardware_type(hardware_type), in OpStatsInfo()
62 HardwareType hardware_type; member
|
| D | op_stats_combiner.cc | 128 bool no_accelerator_in_system, int src_host_id, HardwareType hardware_type, in CombineOpStats() argument
140 if (!IsCoordinator(no_accelerator_in_system, hardware_type)) { in CombineOpStats()
170 bool IsCoordinator(bool no_accelerator_in_system, HardwareType hardware_type) { in IsCoordinator() argument
175 return !HasDevice(hardware_type) && !no_accelerator_in_system; in IsCoordinator()
180 if (HasDevice(op_stats_info.hardware_type)) { in NoAcceleratorInSystem()
199 if (IsCoordinator(no_accelerator_in_system, op_stats_info.hardware_type)) in ComputeStepIntersectionToMergeOpStats()
244 op_stats_info.hardware_type, step_intersection, in CombineAllOpStats()
|
| D | op_stats_to_overview_page.cc | 83 void ComputeDeviceTips(HardwareType hardware_type, in ComputeDeviceTips() argument
85 absl::string_view device_name = HardwareType_Name(hardware_type); in ComputeDeviceTips()
88 if (hardware_type == tensorflow::profiler::TPU) { in ComputeDeviceTips()
137 absl::string_view output_statement, HardwareType hardware_type, in SetCommonRecommendation() argument
150 ComputeDeviceTips(hardware_type, re); in SetCommonRecommendation()
|
| D | op_stats_to_overview_page.h | 47 absl::string_view output_statement, HardwareType hardware_type,
|
| /external/tensorflow/tensorflow/core/profiler/protobuf/ |
| D | input_pipeline.proto | 153 string hardware_type = 9; field
|