{"entrypoint":"main","feature_set":["functions","modules","lexical_scope","checked_int64","finite_float64","bounded_loop","list","record","tuple","option","result","tensor","observer","recovery","policy_effect","checkpoint","record_annotations","type_aliases","contextual_empty_lists"],"integrity_sha256":"1525e4acb1403f1907fb5446ba66642ab793ac5a6052ba082f28a5aa2529ac4e","limits":{"max_items":100000,"max_output":1000000,"max_stack":128,"max_steps":100000,"max_tensor":4096},"module_graph":[{"imports":[],"module":"model_graph","source_sha256":"d6b00095f8850cc1a8dc5f1d7fd0838e5c30faa98a8943c17aa4c95efa974574"}],"numeric_policy":{"atol":1e-06,"float64":"finite_only","int64":"checked_signed","integer_division":"truncate_toward_zero","normalize_epsilon":1e-05,"rtol":1e-05},"observer_context":{"implicit_conversion":false,"mode":"explicit_metadata"},"operand_types":[["Tensor[1,30]","Tensor[30,32]"],["Tensor[1,32]","Tensor[1,32]"],["Tensor[1,32]"],["Tensor[1,32]","Tensor[32,32]"],["Tensor[1,32]","Tensor[1,32]"],["Tensor[1,32]"],["Tensor[1,32]","Tensor[32,22]"],["Tensor[1,22]","Tensor[1,22]"]],"ops":[{"name":"matmul","operand_types":["Tensor[1,30]","Tensor[30,32]"],"result_type":"Tensor[1,32]"},{"name":"add","operand_types":["Tensor[1,32]","Tensor[1,32]"],"result_type":"Tensor[1,32]"},{"name":"tanh","operand_types":["Tensor[1,32]"],"result_type":"Tensor[1,32]"},{"name":"matmul","operand_types":["Tensor[1,32]","Tensor[32,32]"],"result_type":"Tensor[1,32]"},{"name":"add","operand_types":["Tensor[1,32]","Tensor[1,32]"],"result_type":"Tensor[1,32]"},{"name":"tanh","operand_types":["Tensor[1,32]"],"result_type":"Tensor[1,32]"},{"name":"matmul","operand_types":["Tensor[1,32]","Tensor[32,22]"],"result_type":"Tensor[1,22]"},{"name":"add","operand_types":["Tensor[1,22]","Tensor[1,22]"],"result_type":"Tensor[1,22]"}],"program_id":"f30a68d253716d362ab399dc0cd016cf2cdcf2cad60560651f4fd4d15d4c1307","provenance_refs":[{"module":"model_graph","source_sha256":"d6b00095f8850cc1a8dc5f1d7fd0838e5c30faa98a8943c17aa4c95efa974574","status":"COMPILED_LOCAL_SOURCE"}],"required_effects":[],"schema_sha256":"0dd381cb61f65f3346c07918aae587c42c583f8b4449583fdd4dca1afdcee60c","schema_version":"0.2.1","shape_constraints":[{"left":[1,30],"op":"matmul","output":[1,32],"right":[30,32]},{"left":[1,32],"op":"matmul","output":[1,32],"right":[32,32]},{"left":[1,32],"op":"matmul","output":[1,22],"right":[32,22]}],"source_sha256":"d6b00095f8850cc1a8dc5f1d7fd0838e5c30faa98a8943c17aa4c95efa974574","typed_blocks":[{"body":[{"expr":{"args":[{"args":[{"args":[{"inferred_type":"Tensor[1,30]","kind":"var","name":"x","span":{"column":24,"line":3,"offset":182}},{"inferred_type":"Tensor[30,32]","kind":"var","name":"w1","span":{"column":26,"line":3,"offset":184}}],"inferred_type":"Tensor[1,32]","kind":"call","name":"matmul","span":{"column":17,"line":3,"offset":175}},{"inferred_type":"Tensor[1,32]","kind":"var","name":"b1","span":{"column":30,"line":3,"offset":188}}],"inferred_type":"Tensor[1,32]","kind":"call","name":"add","span":{"column":13,"line":3,"offset":171}}],"inferred_type":"Tensor[1,32]","kind":"call","name":"tanh","span":{"column":8,"line":3,"offset":166}},"inferred_type":"Tensor[1,32]","kind":"let","mutable":false,"name":"h1","span":{"column":1,"line":3,"offset":159},"type":null},{"expr":{"args":[{"args":[{"args":[{"inferred_type":"Tensor[1,32]","kind":"var","name":"h1","span":{"column":24,"line":4,"offset":217}},{"inferred_type":"Tensor[32,32]","kind":"var","name":"w2","span":{"column":27,"line":4,"offset":220}}],"inferred_type":"Tensor[1,32]","kind":"call","name":"matmul","span":{"column":17,"line":4,"offset":210}},{"inferred_type":"Tensor[1,32]","kind":"var","name":"b2","span":{"column":31,"line":4,"offset":224}}],"inferred_type":"Tensor[1,32]","kind":"call","name":"add","span":{"column":13,"line":4,"offset":206}}],"inferred_type":"Tensor[1,32]","kind":"call","name":"tanh","span":{"column":8,"line":4,"offset":201}},"inferred_type":"Tensor[1,32]","kind":"let","mutable":false,"name":"h2","span":{"column":1,"line":4,"offset":194},"type":null},{"expr":{"args":[{"args":[{"inferred_type":"Tensor[1,32]","kind":"var","name":"h2","span":{"column":19,"line":5,"offset":248}},{"inferred_type":"Tensor[32,22]","kind":"var","name":"wo","span":{"column":22,"line":5,"offset":251}}],"inferred_type":"Tensor[1,22]","kind":"call","name":"matmul","span":{"column":12,"line":5,"offset":241}},{"inferred_type":"Tensor[1,22]","kind":"var","name":"bo","span":{"column":26,"line":5,"offset":255}}],"inferred_type":"Tensor[1,22]","kind":"call","name":"add","span":{"column":8,"line":5,"offset":237}},"kind":"return","span":{"column":1,"line":5,"offset":230}}],"name":"main","params":[["x","Tensor[1,30]"],["w1","Tensor[30,32]"],["b1","Tensor[1,32]"],["w2","Tensor[32,32]"],["b2","Tensor[1,32]"],["wo","Tensor[32,22]"],["bo","Tensor[1,22]"]],"return_type":"Tensor[1,22]","span":{"column":1,"line":2,"offset":20}}]}