Searched defs:num_gpus_ (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/optimizers/ | ||
| D | auto_parallel.h | 52 int num_gpus_; variable |
| /external/tensorflow/tensorflow/core/profiler/internal/gpu/ | ||
| D | cupti_tracer.h | 120 int num_gpus_; variable |
| D | cupti_collector.cc | 726 int num_gpus_; member in tensorflow::profiler::CuptiTraceCollectorImpl |