Lines Matching defs:num_chunks
58 // information (e.g., timestamp, device_name, num_chunks, chunk_index, dtype,
62 const uint64 wall_time_us, const size_t num_chunks,
84 metadata.set_num_chunks(num_chunks);
156 const size_t num_chunks = cutoffs.size();
158 for (size_t i = 0; i < num_chunks; ++i) {
160 num_chunks, i, tensor_proto->dtype(),
205 const size_t num_chunks =
209 for (size_t i = 0; i < num_chunks; ++i) {
212 (i == num_chunks - 1) ? (total_length - pos) : chunk_size_ub;
214 num_chunks, i, tensor_proto.dtype(),
253 const size_t num_chunks =
256 for (size_t i = 0; i < num_chunks; ++i) {
258 const size_t len = (i == num_chunks - 1)
264 // <hash64>,<device_name>,<wall_time>|<index>|<num_chunks>|.
265 // TODO(cais): Use DebuggerEventMetadata to store device_name, num_chunks
268 "|", i, "|", num_chunks, "|",
271 event, debug_url, num_chunks - 1 == i);
274 "Failed to send chunk ", i, " of ", num_chunks,
