HomeSort by relevance Sort by last modified time
    Searched defs:computation_count (Results 1 - 4 of 4) sorted by null

  /external/tensorflow/tensorflow/compiler/xla/service/
computation_placer.h 41 DeviceAssignment(int replica_count, int computation_count)
42 : Array2D<int>(replica_count, computation_count, -1) {
44 CHECK_GT(computation_count, 0);
48 int computation_count() const { return width(); } function in class:xla::DeviceAssignment
68 // instance for [replica_count x computation_count] setup. The returned device
71 int replica_count, int computation_count);
76 int computation_count);
hlo_execution_profile.h 57 size_t computation_count() const { function in class:xla::HloProfileIndexMap
62 return instruction_count() + computation_count();
hlo_module.h 135 int64 computation_count() const { return computations_.size(); } function in class:xla::HloModule
hlo_cost_analysis.cc 89 auto computation_count = ShapeUtil::ElementsIn(shape); local
96 current_properties_[kTranscendentalsKey] = computation_count;
100 current_properties_[kFlopsKey] = computation_count;

Completed in 272 milliseconds