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()
44 DeviceAssignment(int replica_count, int computation_count) in DeviceAssignment()51 int computation_count() const { return width(); } in computation_count() function
126 int computation_count) { in DeviceId()134 int replica_count, int computation_count) { in AssignDevices()
65 size_t computation_count() const { in computation_count() function
183 int64 computation_count() const { return computations_.size(); } in computation_count() function
98 auto computation_count = ShapeUtil::ElementsIn(shape); in HandleElementwiseOp() local
379 int32 computation_count = 2; field