1dict_value {
2  fields {
3    key: "collect_data_spec"
4    value {
5      named_tuple_value {
6        name: "Trajectory"
7        values {
8          key: "step_type"
9          value {
10            tensor_spec_value {
11              name: "step_type"
12              shape {
13              }
14              dtype: DT_INT32
15            }
16          }
17        }
18        values {
19          key: "observation"
20          value {
21            dict_value {
22              fields {
23                key: "end_bb_freq_by_max"
24                value {
25                  tensor_spec_value {
26                    name: "end_bb_freq_by_max"
27                    shape {
28                      dim {
29                        size: 33
30                      }
31                    }
32                    dtype: DT_FLOAT
33                  }
34                }
35              }
36              fields {
37                key: "hint_weights_by_max"
38                value {
39                  tensor_spec_value {
40                    name: "hint_weights_by_max"
41                    shape {
42                      dim {
43                        size: 33
44                      }
45                    }
46                    dtype: DT_FLOAT
47                  }
48                }
49              }
50              fields {
51                key: "hottest_bb_freq_by_max"
52                value {
53                  tensor_spec_value {
54                    name: "hottest_bb_freq_by_max"
55                    shape {
56                      dim {
57                        size: 33
58                      }
59                    }
60                    dtype: DT_FLOAT
61                  }
62                }
63              }
64              fields {
65                key: "is_free"
66                value {
67                  tensor_spec_value {
68                    name: "is_free"
69                    shape {
70                      dim {
71                        size: 33
72                      }
73                    }
74                    dtype: DT_INT64
75                  }
76                }
77              }
78              fields {
79                key: "is_hint"
80                value {
81                  tensor_spec_value {
82                    name: "is_hint"
83                    shape {
84                      dim {
85                        size: 33
86                      }
87                    }
88                    dtype: DT_INT64
89                  }
90                }
91              }
92              fields {
93                key: "is_local"
94                value {
95                  tensor_spec_value {
96                    name: "is_local"
97                    shape {
98                      dim {
99                        size: 33
100                      }
101                    }
102                    dtype: DT_INT64
103                  }
104                }
105              }
106              fields {
107                key: "liverange_size"
108                value {
109                  tensor_spec_value {
110                    name: "liverange_size"
111                    shape {
112                      dim {
113                        size: 33
114                      }
115                    }
116                    dtype: DT_FLOAT
117                  }
118                }
119              }
120              fields {
121                key: "mask"
122                value {
123                  tensor_spec_value {
124                    name: "mask"
125                    shape {
126                      dim {
127                        size: 33
128                      }
129                    }
130                    dtype: DT_INT64
131                  }
132                }
133              }
134              fields {
135                key: "max_stage"
136                value {
137                  bounded_tensor_spec_value {
138                    name: "max_stage"
139                    shape {
140                      dim {
141                        size: 33
142                      }
143                    }
144                    dtype: DT_INT64
145                    minimum {
146                      dtype: DT_INT64
147                      tensor_shape {
148                      }
149                      int64_val: 0
150                    }
151                    maximum {
152                      dtype: DT_INT64
153                      tensor_shape {
154                      }
155                      int64_val: 6
156                    }
157                  }
158                }
159              }
160              fields {
161                key: "min_stage"
162                value {
163                  bounded_tensor_spec_value {
164                    name: "min_stage"
165                    shape {
166                      dim {
167                        size: 33
168                      }
169                    }
170                    dtype: DT_INT64
171                    minimum {
172                      dtype: DT_INT64
173                      tensor_shape {
174                      }
175                      int64_val: 0
176                    }
177                    maximum {
178                      dtype: DT_INT64
179                      tensor_shape {
180                      }
181                      int64_val: 6
182                    }
183                  }
184                }
185              }
186              fields {
187                key: "nr_broken_hints"
188                value {
189                  tensor_spec_value {
190                    name: "nr_broken_hints"
191                    shape {
192                      dim {
193                        size: 33
194                      }
195                    }
196                    dtype: DT_FLOAT
197                  }
198                }
199              }
200              fields {
201                key: "nr_defs_and_uses"
202                value {
203                  tensor_spec_value {
204                    name: "nr_defs_and_uses"
205                    shape {
206                      dim {
207                        size: 33
208                      }
209                    }
210                    dtype: DT_FLOAT
211                  }
212                }
213              }
214              fields {
215                key: "nr_rematerializable"
216                value {
217                  tensor_spec_value {
218                    name: "nr_rematerializable"
219                    shape {
220                      dim {
221                        size: 33
222                      }
223                    }
224                    dtype: DT_FLOAT
225                  }
226                }
227              }
228              fields {
229                key: "nr_urgent"
230                value {
231                  tensor_spec_value {
232                    name: "nr_urgent"
233                    shape {
234                      dim {
235                        size: 33
236                      }
237                    }
238                    dtype: DT_FLOAT
239                  }
240                }
241              }
242              fields {
243                key: "progress"
244                value {
245                  bounded_tensor_spec_value {
246                    name: "progress"
247                    shape {
248                    }
249                    dtype: DT_FLOAT
250                    minimum {
251                      dtype: DT_FLOAT
252                      tensor_shape {
253                      }
254                      float_val: 0.0
255                    }
256                    maximum {
257                      dtype: DT_FLOAT
258                      tensor_shape {
259                      }
260                      float_val: 1.0
261                    }
262                  }
263                }
264              }
265              fields {
266                key: "start_bb_freq_by_max"
267                value {
268                  tensor_spec_value {
269                    name: "start_bb_freq_by_max"
270                    shape {
271                      dim {
272                        size: 33
273                      }
274                    }
275                    dtype: DT_FLOAT
276                  }
277                }
278              }
279              fields {
280                key: "use_def_density"
281                value {
282                  tensor_spec_value {
283                    name: "use_def_density"
284                    shape {
285                      dim {
286                        size: 33
287                      }
288                    }
289                    dtype: DT_FLOAT
290                  }
291                }
292              }
293              fields {
294                key: "weighed_indvars_by_max"
295                value {
296                  tensor_spec_value {
297                    name: "weighed_indvars_by_max"
298                    shape {
299                      dim {
300                        size: 33
301                      }
302                    }
303                    dtype: DT_FLOAT
304                  }
305                }
306              }
307              fields {
308                key: "weighed_read_writes_by_max"
309                value {
310                  tensor_spec_value {
311                    name: "weighed_read_writes_by_max"
312                    shape {
313                      dim {
314                        size: 33
315                      }
316                    }
317                    dtype: DT_FLOAT
318                  }
319                }
320              }
321              fields {
322                key: "weighed_reads_by_max"
323                value {
324                  tensor_spec_value {
325                    name: "weighed_reads_by_max"
326                    shape {
327                      dim {
328                        size: 33
329                      }
330                    }
331                    dtype: DT_FLOAT
332                  }
333                }
334              }
335              fields {
336                key: "weighed_writes_by_max"
337                value {
338                  tensor_spec_value {
339                    name: "weighed_writes_by_max"
340                    shape {
341                      dim {
342                        size: 33
343                      }
344                    }
345                    dtype: DT_FLOAT
346                  }
347                }
348              }
349            }
350          }
351        }
352        values {
353          key: "action"
354          value {
355            bounded_tensor_spec_value {
356              name: "index_to_evict"
357              shape {
358              }
359              dtype: DT_INT64
360              minimum {
361                dtype: DT_INT64
362                tensor_shape {
363                }
364                int64_val: 0
365              }
366              maximum {
367                dtype: DT_INT64
368                tensor_shape {
369                }
370                int64_val: 32
371              }
372            }
373          }
374        }
375        values {
376          key: "policy_info"
377          value {
378            tuple_value {
379            }
380          }
381        }
382        values {
383          key: "next_step_type"
384          value {
385            tensor_spec_value {
386              name: "step_type"
387              shape {
388              }
389              dtype: DT_INT32
390            }
391          }
392        }
393        values {
394          key: "reward"
395          value {
396            tensor_spec_value {
397              name: "reward"
398              shape {
399              }
400              dtype: DT_FLOAT
401            }
402          }
403        }
404        values {
405          key: "discount"
406          value {
407            bounded_tensor_spec_value {
408              name: "discount"
409              shape {
410              }
411              dtype: DT_FLOAT
412              minimum {
413                dtype: DT_FLOAT
414                tensor_shape {
415                }
416                float_val: 0.0
417              }
418              maximum {
419                dtype: DT_FLOAT
420                tensor_shape {
421                }
422                float_val: 1.0
423              }
424            }
425          }
426        }
427      }
428    }
429  }
430  fields {
431    key: "policy_state_spec"
432    value {
433      tuple_value {
434      }
435    }
436  }
437}
438