Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/grappler/utils/
Dfunctions.cc62 for (const OutputArgInstantiation& output_arg : output_args_) { in GrapplerFunctionItem() local
448 for (const OutputArgInstantiation& output_arg : item.outputs()) { in Initialize() local
549 for (const OutputArgInstantiation& output_arg : item.outputs()) { in MakeFunctionDef() local
/external/tensorflow/tensorflow/core/framework/
Dop_def.proto59 repeated ArgDef output_arg = 3; field
/external/tensorflow/tensorflow/core/grappler/optimizers/data/
Dnoop_elimination.cc70 bool IsOutputIdentityOfInput(const FunctionDef& fdef, const string& output_arg, in IsOutputIdentityOfInput()
Dfusion_utils_test.cc44 for (const auto &output_arg : function.signature().output_arg()) in CheckUniqueNames() local
Dslack_test.cc120 auto *output_arg = fdef->mutable_signature()->add_output_arg(); in TEST() local
Dmap_vectorization.cc108 const auto& output_arg = vectorized_func->signature().output_arg(i); in CreateMapDefunWrapper() local
356 auto* output_arg = signature->add_output_arg(); in AddBranch() local
Dvectorization_utils_test.cc66 for (const auto& output_arg : inner.signature().output_arg()) { in WrapFunctionWithMapDefun() local
Dfusion_utils.cc172 for (const auto& output_arg : second_signature.output_arg()) { in GetUniqueSignature() local
/external/tensorflow/tensorflow/core/kernels/data/
Drewrite_utils.cc73 for (const auto& output_arg : function_def->signature().output_arg()) { in RemoveFakeSinks() local
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dfunction_api_info.cc53 for (const auto& output_arg : function_def.signature().output_arg()) { in Init() local
Dconstant_folding.cc1104 for (const OpDef::ArgDef& output_arg : op_def->output_arg()) { in MaybeFoldable() local
/external/tensorflow/tensorflow/core/api_def/
Dupdate_api_def.cc51 for (auto& output_arg : op.output_arg()) { in FillBaseApiDef() local
/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/c/
Dc_api_unified_experimental_mlir.cc349 for (const tensorflow::OpDef::ArgDef& output_arg : op_def_->output_arg()) { in Create() local
/external/tensorflow/tensorflow/core/distributed_runtime/eager/
Deager_service_impl.cc70 for (const auto& output_arg : op_def.output_arg()) { in GetNumRetvals() local
/external/tensorflow/tensorflow/compiler/tf2xla/
Dtf2xla_util.cc280 const OpDef_ArgDef& output_arg = body_func->signature().output_arg(i); in PropagateConstIntoWhileNode() local
Dxla_compiler.cc620 for (const auto& output_arg : n->op_def().output_arg()) { in GetGraph() local
/external/tensorflow/tensorflow/compiler/jit/
Dencapsulate_subgraphs_pass.cc1238 for (const auto& output_arg : n->op_def().output_arg()) { in Run() local
/external/tensorflow/tensorflow/core/grappler/costs/
Dgraph_properties.cc862 for (const auto& output_arg : grappler_function_item.outputs()) { in UpdateFunction() local
901 for (const auto& output_arg : grappler_function_item.outputs()) { in UpdateFunction() local
/external/tensorflow/tensorflow/python/eager/
Dpywrap_tfe_src.cc3829 const auto& output_arg = op_def->output_arg(i); local