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: "call_argument_setup"
24                value {
25                  tensor_spec_value {
26                    name: "call_argument_setup"
27                    shape {
28                    }
29                    dtype: DT_INT64
30                  }
31                }
32              }
33              fields {
34                key: "call_penalty"
35                value {
36                  tensor_spec_value {
37                    name: "call_penalty"
38                    shape {
39                    }
40                    dtype: DT_INT64
41                  }
42                }
43              }
44              fields {
45                key: "callee_basic_block_count"
46                value {
47                  tensor_spec_value {
48                    name: "callee_basic_block_count"
49                    shape {
50                    }
51                    dtype: DT_INT64
52                  }
53                }
54              }
55              fields {
56                key: "callee_conditionally_executed_blocks"
57                value {
58                  tensor_spec_value {
59                    name: "callee_conditionally_executed_blocks"
60                    shape {
61                    }
62                    dtype: DT_INT64
63                  }
64                }
65              }
66              fields {
67                key: "callee_users"
68                value {
69                  tensor_spec_value {
70                    name: "callee_users"
71                    shape {
72                    }
73                    dtype: DT_INT64
74                  }
75                }
76              }
77              fields {
78                key: "caller_basic_block_count"
79                value {
80                  tensor_spec_value {
81                    name: "caller_basic_block_count"
82                    shape {
83                    }
84                    dtype: DT_INT64
85                  }
86                }
87              }
88              fields {
89                key: "caller_conditionally_executed_blocks"
90                value {
91                  tensor_spec_value {
92                    name: "caller_conditionally_executed_blocks"
93                    shape {
94                    }
95                    dtype: DT_INT64
96                  }
97                }
98              }
99              fields {
100                key: "caller_users"
101                value {
102                  tensor_spec_value {
103                    name: "caller_users"
104                    shape {
105                    }
106                    dtype: DT_INT64
107                  }
108                }
109              }
110              fields {
111                key: "callsite_cost"
112                value {
113                  tensor_spec_value {
114                    name: "callsite_cost"
115                    shape {
116                    }
117                    dtype: DT_INT64
118                  }
119                }
120              }
121              fields {
122                key: "callsite_height"
123                value {
124                  tensor_spec_value {
125                    name: "callsite_height"
126                    shape {
127                    }
128                    dtype: DT_INT64
129                  }
130                }
131              }
132              fields {
133                key: "case_cluster_penalty"
134                value {
135                  tensor_spec_value {
136                    name: "case_cluster_penalty"
137                    shape {
138                    }
139                    dtype: DT_INT64
140                  }
141                }
142              }
143              fields {
144                key: "cold_cc_penalty"
145                value {
146                  tensor_spec_value {
147                    name: "cold_cc_penalty"
148                    shape {
149                    }
150                    dtype: DT_INT64
151                  }
152                }
153              }
154              fields {
155                key: "constant_args"
156                value {
157                  tensor_spec_value {
158                    name: "constant_args"
159                    shape {
160                    }
161                    dtype: DT_INT64
162                  }
163                }
164              }
165              fields {
166                key: "constant_offset_ptr_args"
167                value {
168                  tensor_spec_value {
169                    name: "constant_offset_ptr_args"
170                    shape {
171                    }
172                    dtype: DT_INT64
173                  }
174                }
175              }
176              fields {
177                key: "cost_estimate"
178                value {
179                  tensor_spec_value {
180                    name: "cost_estimate"
181                    shape {
182                    }
183                    dtype: DT_INT64
184                  }
185                }
186              }
187              fields {
188                key: "dead_blocks"
189                value {
190                  tensor_spec_value {
191                    name: "dead_blocks"
192                    shape {
193                    }
194                    dtype: DT_INT64
195                  }
196                }
197              }
198              fields {
199                key: "edge_count"
200                value {
201                  tensor_spec_value {
202                    name: "edge_count"
203                    shape {
204                    }
205                    dtype: DT_INT64
206                  }
207                }
208              }
209              fields {
210                key: "indirect_call_penalty"
211                value {
212                  tensor_spec_value {
213                    name: "indirect_call_penalty"
214                    shape {
215                    }
216                    dtype: DT_INT64
217                  }
218                }
219              }
220              fields {
221                key: "inlining_default"
222                value {
223                  tensor_spec_value {
224                    name: "inlining_default"
225                    shape {
226                    }
227                    dtype: DT_INT64
228                  }
229                }
230              }
231              fields {
232                key: "is_multiple_blocks"
233                value {
234                  tensor_spec_value {
235                    name: "is_multiple_blocks"
236                    shape {
237                    }
238                    dtype: DT_INT64
239                  }
240                }
241              }
242              fields {
243                key: "jump_table_penalty"
244                value {
245                  tensor_spec_value {
246                    name: "jump_table_penalty"
247                    shape {
248                    }
249                    dtype: DT_INT64
250                  }
251                }
252              }
253              fields {
254                key: "last_call_to_static_bonus"
255                value {
256                  tensor_spec_value {
257                    name: "last_call_to_static_bonus"
258                    shape {
259                    }
260                    dtype: DT_INT64
261                  }
262                }
263              }
264              fields {
265                key: "load_elimination"
266                value {
267                  tensor_spec_value {
268                    name: "load_elimination"
269                    shape {
270                    }
271                    dtype: DT_INT64
272                  }
273                }
274              }
275              fields {
276                key: "load_relative_intrinsic"
277                value {
278                  tensor_spec_value {
279                    name: "load_relative_intrinsic"
280                    shape {
281                    }
282                    dtype: DT_INT64
283                  }
284                }
285              }
286              fields {
287                key: "lowered_call_arg_setup"
288                value {
289                  tensor_spec_value {
290                    name: "lowered_call_arg_setup"
291                    shape {
292                    }
293                    dtype: DT_INT64
294                  }
295                }
296              }
297              fields {
298                key: "nested_inline_cost_estimate"
299                value {
300                  tensor_spec_value {
301                    name: "nested_inline_cost_estimate"
302                    shape {
303                    }
304                    dtype: DT_INT64
305                  }
306                }
307              }
308              fields {
309                key: "nested_inlines"
310                value {
311                  tensor_spec_value {
312                    name: "nested_inlines"
313                    shape {
314                    }
315                    dtype: DT_INT64
316                  }
317                }
318              }
319              fields {
320                key: "node_count"
321                value {
322                  tensor_spec_value {
323                    name: "node_count"
324                    shape {
325                    }
326                    dtype: DT_INT64
327                  }
328                }
329              }
330              fields {
331                key: "nr_ctant_params"
332                value {
333                  tensor_spec_value {
334                    name: "nr_ctant_params"
335                    shape {
336                    }
337                    dtype: DT_INT64
338                  }
339                }
340              }
341              fields {
342                key: "num_loops"
343                value {
344                  tensor_spec_value {
345                    name: "num_loops"
346                    shape {
347                    }
348                    dtype: DT_INT64
349                  }
350                }
351              }
352              fields {
353                key: "simplified_instructions"
354                value {
355                  tensor_spec_value {
356                    name: "simplified_instructions"
357                    shape {
358                    }
359                    dtype: DT_INT64
360                  }
361                }
362              }
363              fields {
364                key: "sroa_losses"
365                value {
366                  tensor_spec_value {
367                    name: "sroa_losses"
368                    shape {
369                    }
370                    dtype: DT_INT64
371                  }
372                }
373              }
374              fields {
375                key: "sroa_savings"
376                value {
377                  tensor_spec_value {
378                    name: "sroa_savings"
379                    shape {
380                    }
381                    dtype: DT_INT64
382                  }
383                }
384              }
385              fields {
386                key: "switch_penalty"
387                value {
388                  tensor_spec_value {
389                    name: "switch_penalty"
390                    shape {
391                    }
392                    dtype: DT_INT64
393                  }
394                }
395              }
396              fields {
397                key: "threshold"
398                value {
399                  tensor_spec_value {
400                    name: "threshold"
401                    shape {
402                    }
403                    dtype: DT_INT64
404                  }
405                }
406              }
407              fields {
408                key: "unsimplified_common_instructions"
409                value {
410                  tensor_spec_value {
411                    name: "unsimplified_common_instructions"
412                    shape {
413                    }
414                    dtype: DT_INT64
415                  }
416                }
417              }
418            }
419          }
420        }
421        values {
422          key: "action"
423          value {
424            bounded_tensor_spec_value {
425              name: "inlining_decision"
426              shape {
427              }
428              dtype: DT_INT64
429              minimum {
430                dtype: DT_INT64
431                tensor_shape {
432                }
433                int64_val: 0
434              }
435              maximum {
436                dtype: DT_INT64
437                tensor_shape {
438                }
439                int64_val: 1
440              }
441            }
442          }
443        }
444        values {
445          key: "policy_info"
446          value {
447            tuple_value {
448            }
449          }
450        }
451        values {
452          key: "next_step_type"
453          value {
454            tensor_spec_value {
455              name: "step_type"
456              shape {
457              }
458              dtype: DT_INT32
459            }
460          }
461        }
462        values {
463          key: "reward"
464          value {
465            tensor_spec_value {
466              name: "reward"
467              shape {
468              }
469              dtype: DT_FLOAT
470            }
471          }
472        }
473        values {
474          key: "discount"
475          value {
476            bounded_tensor_spec_value {
477              name: "discount"
478              shape {
479              }
480              dtype: DT_FLOAT
481              minimum {
482                dtype: DT_FLOAT
483                tensor_shape {
484                }
485                float_val: 0.0
486              }
487              maximum {
488                dtype: DT_FLOAT
489                tensor_shape {
490                }
491                float_val: 1.0
492              }
493            }
494          }
495        }
496      }
497    }
498  }
499  fields {
500    key: "policy_state_spec"
501    value {
502      tuple_value {
503      }
504    }
505  }
506}
507