Searched refs:api_def_arg (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/js/ops/ |
| D | ts_op_gen.cc | 34 ArgDefs(const OpDef::ArgDef& op_def_arg, const ApiDef::Arg& api_def_arg) in ArgDefs()
35 : op_def_arg(op_def_arg), api_def_arg(api_def_arg) {} in ArgDefs()
38 const ApiDef::Arg& api_def_arg; member
115 auto api_def_arg = FindInputArg(api_def_.arg_order(i), api_def_); in ProcessArgs() local
116 if (api_def_arg == nullptr) { in ProcessArgs()
132 input_op_args_.push_back(ArgDefs(*op_def_arg, *api_def_arg)); in ProcessArgs()
|
| /external/tensorflow/tensorflow/core/api_def/ |
| D | api_test.cc | 90 for (const auto& api_def_arg : api_def.in_arg()) { in TestAllApiDefInputArgsAreValid() local
93 if (api_def_arg.name() == op_arg.name()) { in TestAllApiDefInputArgsAreValid()
99 << "Input argument " << api_def_arg.name() in TestAllApiDefInputArgsAreValid()
114 for (const auto& api_def_arg : api_def.out_arg()) { in TestAllApiDefOutputArgsAreValid() local
117 if (api_def_arg.name() == op_arg.name()) { in TestAllApiDefOutputArgsAreValid()
123 << "Output argument " << api_def_arg.name() in TestAllApiDefOutputArgsAreValid()
|
| /external/tensorflow/tensorflow/cc/framework/ |
| D | cc_op_gen.cc | 572 const auto& api_def_arg = *FindInputArg(api_def.arg_order(i), api_def); in OpInfo() local
575 arg_names.push_back(AvoidCPPKeywords(api_def_arg.rename_to())); in OpInfo()
578 StringPiece description = api_def_arg.description(); in OpInfo()
582 AvoidCPPKeywords(api_def_arg.rename_to()), ": ", in OpInfo()
583 api_def_arg.description(), "\n"); in OpInfo()
633 const auto& api_def_arg(api_def.out_arg(i)); in OpInfo() local
634 CHECK_EQ(arg.name(), api_def_arg.name()); in OpInfo()
639 output_names.push_back(AvoidCPPKeywords(api_def_arg.rename_to())); in OpInfo()
933 const auto& api_def_arg(api_def.in_arg(i)); in GetConstructorBody() local
935 &body, " auto _", api_def_arg.rename_to(), " = ::tensorflow::ops::", in GetConstructorBody()
[all …]
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | python_op_gen_internal.cc | 515 const auto& api_def_arg = *FindInputArg(api_def_.arg_order(i), api_def_); in Code() local
516 params_no_default.emplace_back(api_def_arg.name(), api_def_arg.rename_to()); in Code()
669 const auto& api_def_arg = *FindInputArg(api_def_.arg_order(i), api_def_); in AddDocStringInputs() local
670 StringPiece description = api_def_arg.description(); in AddDocStringInputs()
|
| D | python_op_gen.cc | 239 const auto& api_def_arg = *FindInputArg(api_def_.arg_order(i), api_def_); in Code() local
240 params_no_default_.emplace_back(api_def_arg.name(), in Code()
241 api_def_arg.rename_to()); in Code()
|