| 318 | } |
| 319 | |
| 320 | ArgumentSpec CreateOutput(const OpDef_ArgDef& output_def, |
| 321 | const ApiDef::Arg& output_api, |
| 322 | TypeResolver* type_resolver) { |
| 323 | bool iterable = false; |
| 324 | Type type = type_resolver->TypeOf(output_def, &iterable); |
| 325 | Type var_type = Type::Class("Output", "org.tensorflow").add_parameter(type); |
| 326 | if (iterable) { |
| 327 | var_type = Type::ListOf(var_type); |
| 328 | } |
| 329 | return ArgumentSpec( |
| 330 | output_api.name(), |
| 331 | Variable::Create(SnakeToCamelCase(output_api.rename_to()), var_type), |
| 332 | type, ParseDocumentation(output_api.description()), iterable); |
| 333 | } |
| 334 | |
| 335 | EndpointSpec CreateEndpoint(const OpDef& op_def, const ApiDef& api_def, |
| 336 | const ApiDef_Endpoint& endpoint_def) { |
no test coverage detected