Home
last modified time | relevance | path

Searched defs:input_arg (Results 1 – 20 of 20) sorted by relevance

/external/tensorflow/tensorflow/core/framework/
Dnode_def_builder.cc115 void NodeDefBuilder::SingleInput(const OpDef::ArgDef* input_arg, in SingleInput()
136 void NodeDefBuilder::ListInput(const OpDef::ArgDef* input_arg, in ListInput()
186 void NodeDefBuilder::VerifyInputType(const OpDef::ArgDef* input_arg, in VerifyInputType()
195 void NodeDefBuilder::VerifyInputRef(const OpDef::ArgDef* input_arg, in VerifyInputRef()
Dnode_def_builder.h175 DataType MaybeAddRef(const OpDef::ArgDef* input_arg, DataType dt) { in MaybeAddRef()
Dop_def.proto56 repeated ArgDef input_arg = 2; field
/external/tensorflow/tensorflow/core/grappler/utils/
Dfunctions.cc58 for (const InputArgInstantiation& input_arg : input_args_) { in GrapplerFunctionItem() local
323 const InputArgInstantiation& input_arg = item->input(input_index); in ReplaceInputWithConst() local
445 for (const InputArgInstantiation& input_arg : item.inputs()) { in Initialize() local
540 for (const InputArgInstantiation& input_arg : item.inputs()) { in MakeFunctionDef() local
/external/tensorflow/tensorflow/core/grappler/optimizers/data/
Dnoop_elimination.cc71 const string& input_arg) { in IsOutputIdentityOfInput()
Dfusion_utils_test.cc39 for (const auto &input_arg : function.signature().input_arg()) in CheckUniqueNames() local
Dslack_test.cc117 auto *input_arg = fdef->mutable_signature()->add_input_arg(); in TEST() local
Dfusion_utils.cc161 for (const auto& input_arg : second_signature.input_arg()) { in GetUniqueSignature() local
387 for (const auto& input_arg : first_function.signature().input_arg()) { in LazyConjunctionNodes() local
Dfunction_utils.cc107 auto* input_arg = fdef->mutable_signature()->mutable_input_arg()->Add(); in AddFunctionInput() local
Dmap_vectorization.cc387 auto* input_arg = signature->add_input_arg(); in AddBranch() local
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dfunction_api_info.cc49 for (const auto& input_arg : function_def.signature().input_arg()) { in Init() local
Darithmetic_optimizer.cc2110 const OpDef::ArgDef& input_arg = op_def->input_arg(0); in SetInputType() local
/external/tensorflow/tensorflow/core/api_def/
Dupdate_api_def.cc44 for (auto& input_arg : op.input_arg()) { in FillBaseApiDef() local
/external/libaom/libaom/examples/
Dnoise_model.c81 static const arg_def_t input_arg = ARG_DEF("-i", "input", 1, "Input filename"); variable
/external/tensorflow/tensorflow/python/framework/
Dfunction.py494 def _create_hash_str(self, input_arg, output_arg, node_def): argument
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/
Dexport_graphdef.cc310 auto input_arg = src.cast<BlockArgument>(); in AddEdgeBetweenNodes() local
Dimport_model.cc1301 auto& input_arg = signature.input_arg(i); in ConvertLibFunction() local
/external/tensorflow/tensorflow/c/
Dc_api_experimental.cc456 const tensorflow::OpDef_ArgDef& input_arg = op_def->input_arg()[input_index]; in TF_GetNumberAttrForOpListInput() local
/external/tensorflow/tensorflow/compiler/tf2xla/
Dtf2xla_util.cc287 const OpDef_ArgDef& input_arg = body_func->signature().input_arg(i); in PropagateConstIntoWhileNode() local
/external/tensorflow/tensorflow/python/eager/
Dpywrap_tfe_src.cc3415 const tensorflow::OpDef::ArgDef& input_arg,
3726 const auto& input_arg = op_def->input_arg(i); local