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 | 99 for (const auto& api_def_arg : api_def.in_arg()) { in TestAllApiDefInputArgsAreValid() local
102 if (api_def_arg.name() == op_arg.name()) { in TestAllApiDefInputArgsAreValid()
108 << "Input argument " << api_def_arg.name() in TestAllApiDefInputArgsAreValid()
123 for (const auto& api_def_arg : api_def.out_arg()) { in TestAllApiDefOutputArgsAreValid() local
126 if (api_def_arg.name() == op_arg.name()) { in TestAllApiDefOutputArgsAreValid()
132 << "Output argument " << api_def_arg.name() in TestAllApiDefOutputArgsAreValid()
|
| /external/tensorflow/tensorflow/cc/framework/ |
| D | cc_op_gen.cc | 594 const auto& api_def_arg = *FindInputArg(api_def.arg_order(i), api_def); in OpInfo() local
597 arg_names.push_back(AvoidCPPKeywords(api_def_arg.rename_to())); in OpInfo()
600 StringPiece description = api_def_arg.description(); in OpInfo()
604 AvoidCPPKeywords(api_def_arg.rename_to()), ": ", in OpInfo()
605 api_def_arg.description(), "\n"); in OpInfo()
655 const auto& api_def_arg(api_def.out_arg(i)); in OpInfo() local
656 CHECK_EQ(arg.name(), api_def_arg.name()); in OpInfo()
661 output_names.push_back(AvoidCPPKeywords(api_def_arg.rename_to())); in OpInfo()
955 const auto& api_def_arg(api_def.in_arg(i)); in GetConstructorBody() local
957 &body, " auto _", api_def_arg.rename_to(), " = ::tensorflow::ops::", in GetConstructorBody()
[all …]
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | python_op_gen_internal.cc | 535 const auto& api_def_arg = *FindInputArg(api_def_.arg_order(i), api_def_); in Code() local
536 params_no_default.emplace_back(api_def_arg.name(), api_def_arg.rename_to()); in Code()
690 const auto& api_def_arg = *FindInputArg(api_def_.arg_order(i), api_def_); in AddDocStringInputs() local
691 StringPiece description = api_def_arg.description(); in AddDocStringInputs()
|
| D | python_op_gen.cc | 280 const auto& api_def_arg = *FindInputArg(api_def_.arg_order(i), api_def_); in Code() local
281 params_no_default_.emplace_back(api_def_arg.name(), in Code()
282 api_def_arg.rename_to()); in Code()
|