1op { 2 name: "TPUCompile" 3 input_arg { 4 name: "dynamic_shapes" 5 type: DT_INT64 6 number_attr: "NumDynamicShapes" 7 } 8 input_arg { 9 name: "guaranteed_constants" 10 type_list_attr: "Tguaranteed_constants" 11 } 12 output_arg { 13 name: "compilation_status" 14 type: DT_STRING 15 } 16 output_arg { 17 name: "program" 18 type: DT_STRING 19 number_attr: "num_computations" 20 } 21 output_arg { 22 name: "may_modify_variables" 23 type: DT_BOOL 24 number_attr: "num_computations" 25 } 26 attr { 27 name: "num_computations" 28 type: "int" 29 has_minimum: true 30 } 31 attr { 32 name: "function" 33 type: "func" 34 } 35 attr { 36 name: "metadata" 37 type: "string" 38 } 39 attr { 40 name: "NumDynamicShapes" 41 type: "int" 42 has_minimum: true 43 } 44 attr { 45 name: "Tguaranteed_constants" 46 type: "list(type)" 47 has_minimum: true 48 } 49 is_stateful: true 50} 51