Home
last modified time | relevance | path

Searched refs:OutputArgInstantiation (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/core/grappler/utils/
Dfunctions.h47 struct OutputArgInstantiation { struct
48 OutputArgInstantiation(string node_name, DataType data_type) in OutputArgInstantiation() argument
71 const std::vector<OutputArgInstantiation>& outputs() const; argument
72 const OutputArgInstantiation& output(int i) const;
101 std::vector<OutputArgInstantiation> output_args,
114 std::vector<OutputArgInstantiation> output_args_;
Dfunctions.cc43 std::vector<OutputArgInstantiation> output_args, in GrapplerFunctionItem()
62 for (const OutputArgInstantiation& output_arg : output_args_) { in GrapplerFunctionItem()
89 const std::vector<OutputArgInstantiation>& GrapplerFunctionItem::outputs() in outputs()
94 const OutputArgInstantiation& GrapplerFunctionItem::output(int i) const { in output()
271 std::vector<OutputArgInstantiation> outputs; in MakeGrapplerFunctionItem()
366 absl::flat_hash_set<const OutputArgInstantiation*> remove_output_args; in RemoveFunctionOutputs()
367 const auto is_remove_output_arg = [&](const OutputArgInstantiation& output) { in RemoveFunctionOutputs()
372 const OutputArgInstantiation& output = item->output(i); in RemoveFunctionOutputs()
448 for (const OutputArgInstantiation& output_arg : item.outputs()) { in Initialize()
549 for (const OutputArgInstantiation& output_arg : item.outputs()) { in MakeFunctionDef()