Home
last modified time | relevance | path

Searched refs:OutputArgExpansion (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/grappler/utils/
Dfunctions.h70 struct OutputArgExpansion { struct
166 const std::vector<OutputArgExpansion>& outputs() const;
167 const OutputArgExpansion& output(int i) const;
194 std::vector<OutputArgExpansion> output_arg_expansions,
204 std::vector<OutputArgExpansion> output_arg_expansions_;
Dfunctions.cc313 std::vector<OutputArgExpansion> output_arg_expansions, in GrapplerFunctionItem()
333 for (const OutputArgExpansion& output_arg : output_arg_expansions_) { in GrapplerFunctionItem()
362 const std::vector<OutputArgExpansion>& GrapplerFunctionItem::outputs() const { in outputs()
366 const OutputArgExpansion& GrapplerFunctionItem::output(int i) const { in output()
601 std::vector<OutputArgExpansion> outputs; in MakeGrapplerFunctionItem()
632 OutputArgExpansion output{/*output_name=*/out.name(), in MakeGrapplerFunctionItem()
738 for (const OutputArgExpansion& out_arg : item->outputs()) { in RemoveFunctionOutputs()
752 absl::flat_hash_set<const OutputArgExpansion*> remove_output_args; in RemoveFunctionOutputs()
753 const auto is_remove_output_arg = [&](const OutputArgExpansion& output) { in RemoveFunctionOutputs()
758 const OutputArgExpansion& output = item->output(i); in RemoveFunctionOutputs()
[all …]
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dfunction_optimizer.cc999 for (const OutputArgExpansion& output_arg : item.outputs()) { in InlinedFunctionOutputsNode()
1042 for (const OutputArgExpansion& output_arg : item.outputs()) { in InlineDirectFunctionCall()
1657 for (const OutputArgExpansion& output_arg : item.outputs()) { in InlineIndirectFunctionCall()
1899 for (const OutputArgExpansion& output : item.outputs()) { in InlineIndirectFunctionCall()