| 283 | } |
| 284 | |
| 285 | ArgumentSpec CreateInput(const OpDef_ArgDef& input_def, |
| 286 | const ApiDef::Arg& input_api_def, |
| 287 | TypeResolver* type_resolver) { |
| 288 | bool iterable = false; |
| 289 | Type type = type_resolver->TypeOf(input_def, &iterable); |
| 290 | Type var_type = |
| 291 | Type::Interface("Operand", "org.tensorflow").add_parameter(type); |
| 292 | if (iterable) { |
| 293 | var_type = Type::IterableOf(var_type); |
| 294 | } |
| 295 | return ArgumentSpec( |
| 296 | input_api_def.name(), |
| 297 | Variable::Create(SnakeToCamelCase(input_api_def.rename_to()), var_type), |
| 298 | type, ParseDocumentation(input_api_def.description()), iterable); |
| 299 | } |
| 300 | |
| 301 | AttributeSpec CreateAttribute(const OpDef_AttrDef& attr_def, |
| 302 | const ApiDef::Attr& attr_api_def, |
no test coverage detected