Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/grappler/utils/
Dfunctions.h38 struct InputArgInstantiation { struct
39 InputArgInstantiation(string node_name, DataType data_type) in InputArgInstantiation() function
67 const std::vector<InputArgInstantiation>& inputs() const;
68 const InputArgInstantiation& input(int i) const;
100 std::vector<InputArgInstantiation> input_args,
113 std::vector<InputArgInstantiation> input_args_;
Dfunctions.cc42 std::vector<InputArgInstantiation> input_args, in GrapplerFunctionItem()
58 for (const InputArgInstantiation& input_arg : input_args_) { in GrapplerFunctionItem()
77 const std::vector<InputArgInstantiation>& GrapplerFunctionItem::inputs() const { in inputs()
81 const InputArgInstantiation& GrapplerFunctionItem::input(int i) const { in input()
262 std::vector<InputArgInstantiation> inputs; in MakeGrapplerFunctionItem()
323 const InputArgInstantiation& input_arg = item->input(input_index); in ReplaceInputWithConst()
445 for (const InputArgInstantiation& input_arg : item.inputs()) { in Initialize()
540 for (const InputArgInstantiation& input_arg : item.inputs()) { in MakeFunctionDef()