Home
last modified time | relevance | path

Searched defs:output_arg (Results 1 – 16 of 16) sorted by relevance

/external/tensorflow/tensorflow/core/framework/
Dop_def.proto55 repeated ArgDef output_arg = 3; field
/external/tensorflow/tensorflow/core/grappler/utils/
Dfunctions.cc333 for (const OutputArgExpansion& output_arg : output_arg_expansions_) { in GrapplerFunctionItem() local
794 for (const OutputArgExpansion& output_arg : item.outputs()) { in MakeFunctionDef() local
824 [&output_tensors](const OutputArgExpansion& output_arg) -> const string& { in MakeFunctionDef()
849 for (const OutputArgExpansion& output_arg : item.outputs()) { in MakeFunctionDef() local
/external/tensorflow/tensorflow/core/grappler/optimizers/data/
Dfusion_utils_test.cc44 for (const auto &output_arg : function.signature().output_arg()) in CheckUniqueNames() local
Dmap_vectorization.cc104 const auto& output_arg = vectorized_func->signature().output_arg(i); in CreateMapDefunWrapper() local
335 auto* output_arg = signature->add_output_arg(); in AddBranch() local
Dfusion_utils.cc171 for (const auto& output_arg : second_signature.output_arg()) { in GetUniqueSignature() local
Dvectorization_utils_test.cc66 for (const auto& output_arg : inner.signature().output_arg()) { in WrapFunctionWithMapDefun() local
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dfunction_api_info.cc53 for (const auto& output_arg : function_def.signature().output_arg()) { in Init() local
Dfunction_optimizer.cc999 for (const OutputArgExpansion& output_arg : item.outputs()) { in InlinedFunctionOutputsNode() local
1042 for (const OutputArgExpansion& output_arg : item.outputs()) { in InlineDirectFunctionCall() local
1657 for (const OutputArgExpansion& output_arg : item.outputs()) { in InlineIndirectFunctionCall() local
/external/tensorflow/tensorflow/core/kernels/data/
Dgraph_rewrite_dataset.cc124 for (const auto& output_arg : function_def->signature().output_arg()) { in RemoveFakeSinks() local
/external/tensorflow/tensorflow/core/api_def/
Dupdate_api_def.cc51 for (auto& output_arg : op.output_arg()) { in FillBaseApiDef() local
/external/tensorflow/tensorflow/core/distributed_runtime/eager/
Deager_service_impl.cc58 for (const auto& output_arg : op_def.output_arg()) { in GetNumRetvals() 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.cc281 const OpDef_ArgDef& output_arg = body_func->signature().output_arg(i); in PropagateConstIntoWhileNode() local
Dxla_compiler.cc527 for (const auto& output_arg : n->op_def().output_arg()) { in GetGraph() local
/external/tensorflow/tensorflow/compiler/jit/
Dencapsulate_subgraphs_pass.cc2598 for (const auto& output_arg : n->op_def().output_arg()) { in Run() local
/external/tensorflow/tensorflow/python/eager/
Dpywrap_tfe_src.cc2751 const auto& output_arg = op_def->output_arg(i); local