Searched defs:computation_count (Results 1 – 6 of 6) sorted by relevance
41 DeviceAssignment(int replica_count, int computation_count) in DeviceAssignment()48 int computation_count() const { return width(); } in computation_count() function
93 int computation_count) { in DeviceId()101 int replica_count, int computation_count) { in AssignDevices()
65 size_t computation_count() const { in computation_count() function
168 int64 computation_count() const { return computations_.size(); } in computation_count() function
90 auto computation_count = ShapeUtil::ElementsIn(shape); in HandleElementwiseOp() local
344 int32 computation_count = 2; field