Searched defs:computation_count (Results 1 – 7 of 7) sorted by relevance
39 int computation_count) { in DeviceId()44 int replica_count, int computation_count) { in AssignDevices()59 int computation_count) { in AssignLocalDevices()
42 DeviceAssignment(int replica_count, int computation_count) in DeviceAssignment()49 int computation_count() const { return width(); } in computation_count() function
140 int computation_count) { in DeviceId()148 int replica_count, int computation_count) { in AssignDevices()
64 size_t computation_count() const { in computation_count() function
242 int64_t computation_count() const { return computations_.size(); } in computation_count() function
98 auto computation_count = ShapeUtil::ElementsIn(shape); in HandleElementwiseOp() local
427 int32 computation_count = 2; field