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
117 int computation_count) { in DeviceId()125 int replica_count, int computation_count) { in AssignDevices()
66 size_t computation_count() const { in computation_count() function
176 int64 computation_count() const { return computations_.size(); } in computation_count() function
98 auto computation_count = ShapeUtil::ElementsIn(shape); in HandleElementwiseOp() local
357 int32 computation_count = 2; field