Home
last modified time | relevance | path

Searched refs:api_def_ (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/js/ops/
Dts_op_gen.cc70 const ApiDef& api_def_; member in tensorflow::__anon8001287d0111::GenTypeScriptOp
90 : op_def_(op_def), api_def_(api_def), num_outputs_(0) {} in GenTypeScriptOp()
108 for (int i = 0; i < api_def_.arg_order_size(); i++) { in ProcessArgs()
109 auto op_def_arg = FindInputArg(api_def_.arg_order(i), op_def_); in ProcessArgs()
112 << api_def_.arg_order(i); in ProcessArgs()
115 auto api_def_arg = FindInputArg(api_def_.arg_order(i), api_def_); in ProcessArgs()
118 << api_def_.arg_order(i); in ProcessArgs()
135 num_outputs_ = api_def_.out_arg_size(); in ProcessArgs()
140 op_attrs_.push_back(OpAttrs(op_def_.attr(i), api_def_.attr(i))); in ProcessAttrs()
168 strings::StrAppend(&result_, "export function ", api_def_.endpoint(0).name(), in AddMethodSignature()
/external/tensorflow/tensorflow/python/framework/
Dpython_op_gen_internal.cc499 api_def_(api_def), in GenPythonOp()
513 for (int i = 0; i < api_def_.arg_order_size(); ++i) { in Code()
514 const auto& arg = *FindInputArg(api_def_.arg_order(i), op_def_); in Code()
515 const auto& api_def_arg = *FindInputArg(api_def_.arg_order(i), api_def_); in Code()
527 for (int i = 0; i < api_def_.attr_size(); ++i) { in Code()
528 const auto& attr(api_def_.attr(i)); in Code()
588 if (api_def_.visibility() != ApiDef::VISIBLE) { in AddExport()
593 !api_def_.deprecation_version() || in AddExport()
594 api_def_.deprecation_version() > kLatestAPIExportVersion; in AddExport()
600 for (const auto& endpoint : api_def_.endpoint()) { in AddExport()
[all …]
Dpython_op_gen.cc233 if (api_def_.visibility() == ApiDef::SKIP) { in Code()
237 for (int i = 0; i < api_def_.arg_order_size(); ++i) { in Code()
238 const auto& arg = *FindInputArg(api_def_.arg_order(i), op_def_); in Code()
239 const auto& api_def_arg = *FindInputArg(api_def_.arg_order(i), api_def_); in Code()
253 const auto& api_def_attr(api_def_.attr(i)); in Code()
363 if (api_def_.visibility() == ApiDef::VISIBLE) { in HandleGraphMode()
412 DCHECK_EQ(op_def_.input_arg(i).name(), api_def_.in_arg(i).name()); in GetEagerNotAllowedError()
413 ref_arg = api_def_.in_arg(i).rename_to(); in GetEagerNotAllowedError()
420 DCHECK_EQ(op_def_.output_arg(i).name(), api_def_.out_arg(i).name()); in GetEagerNotAllowedError()
421 ref_arg = api_def_.out_arg(i).rename_to(); in GetEagerNotAllowedError()
[all …]
Dpython_op_gen_internal.h97 const ApiDef& api_def_; variable