Lines Matching refs:num_chunks
55 // informaion (e.g., timestmap, device_name, num_chunks, chunk_index, dtype,
59 const uint64 wall_time_us, const size_t num_chunks,
81 metadata.set_num_chunks(num_chunks);
153 const size_t num_chunks = cutoffs.size();
155 for (size_t i = 0; i < num_chunks; ++i) {
157 num_chunks, i, tensor_proto->dtype(),
202 const size_t num_chunks =
206 for (size_t i = 0; i < num_chunks; ++i) {
209 (i == num_chunks - 1) ? (total_length - pos) : chunk_size_ub;
211 num_chunks, i, tensor_proto.dtype(),
250 const size_t num_chunks =
253 for (size_t i = 0; i < num_chunks; ++i) {
255 const size_t len = (i == num_chunks - 1)
261 // <hash64>,<device_name>,<wall_time>|<index>|<num_chunks>|.
262 // TODO(cais): Use DebuggerEventMetadata to store device_name, num_chunks
265 "|", i, "|", num_chunks, "|",
268 event, debug_url, num_chunks - 1 == i);
271 "Failed to send chunk ", i, " of ", num_chunks,
