Home
last modified time | relevance | path

Searched defs:input_arg (Results 1 – 17 of 17) 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.h169 DataType MaybeAddRef(const OpDef::ArgDef* input_arg, DataType dt) { in MaybeAddRef()
Dop_def.proto52 repeated ArgDef input_arg = 2; field
/external/tensorflow/tensorflow/core/grappler/utils/
Dfunctions.cc147 auto input_arg = input_arg_expansions_.find(node_name); in ExpandFunctionDefInput() local
327 for (const InputArgExpansion& input_arg : input_arg_expansions_) { in GrapplerFunctionItem() local
785 for (const InputArgExpansion& input_arg : item.inputs()) { in MakeFunctionDef() local
837 for (const InputArgExpansion& input_arg : item.inputs()) { in MakeFunctionDef() local
/external/tensorflow/tensorflow/core/grappler/optimizers/data/
Dfusion_utils_test.cc39 for (const auto &input_arg : function.signature().input_arg()) in CheckUniqueNames() local
Dfusion_utils.cc160 for (const auto& input_arg : second_signature.input_arg()) { in GetUniqueSignature() local
386 for (const auto& input_arg : first_function.signature().input_arg()) { in LazyConjunctionNodes() local
Dmap_vectorization.cc366 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
Dfunction_optimizer.cc977 for (const InputArgExpansion& input_arg : item.inputs()) { in InlinedFunctionInputsNode() local
1033 for (const InputArgExpansion& input_arg : item.inputs()) { in InlineDirectFunctionCall() local
1636 for (const InputArgExpansion& input_arg : item.inputs()) { in InlineIndirectFunctionCall() local
Darithmetic_optimizer.cc2007 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/compiler/xla/tests/
Dreduce_window_test.cc623 auto input_arg = CreateParameterAndTransferLiteral(0, input_literal, "p0", in DoIt() local
1167 auto input_arg = CreateParameterAndTransferLiteral(0, input_literal, "p0", in DoIt() local
1360 auto input_arg = in TEST_P() local
/external/tensorflow/tensorflow/c/
Dc_api_experimental.cc8874 const tensorflow::OpDef_ArgDef& input_arg = op_def->input_arg()[input_index]; in TF_GetNumberAttrForOpListInput() local
9079 for (const tensorflow::OpDef::ArgDef& input_arg : op_def.input_arg()) { in TFE_AddEagerOpToGraph() local
/external/tensorflow/tensorflow/python/framework/
Dfunction.py452 def _create_hash_str(self, input_arg, output_arg, node_def): argument
/external/tensorflow/tensorflow/compiler/tf2xla/
Dtf2xla_util.cc288 const OpDef_ArgDef& input_arg = body_func->signature().input_arg(i); in PropagateConstIntoWhileNode() local
/external/tensorflow/tensorflow/python/eager/
Dpywrap_tfe_src.cc2354 const tensorflow::OpDef::ArgDef& input_arg,
2652 const auto& input_arg = op_def->input_arg(i); local