Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/grappler/utils/
Dfunctions.h57 struct InputArgExpansion { struct
90 void RegisterInputArgExpansion(InputArgExpansion input_arg_expansion);
120 absl::flat_hash_map<string, InputArgExpansion> input_arg_expansions_;
162 const std::vector<InputArgExpansion>& inputs() const;
163 const InputArgExpansion& input(int i) const;
193 std::vector<InputArgExpansion> input_arg_expansions,
203 std::vector<InputArgExpansion> input_arg_expansions_;
Dfunctions.cc80 InputArgExpansion input_arg_expansion) { in RegisterInputArgExpansion()
149 const InputArgExpansion& input_arg_expansion = input_arg->second; in ExpandFunctionDefInput()
312 std::vector<InputArgExpansion> input_arg_expansions, in GrapplerFunctionItem()
327 for (const InputArgExpansion& input_arg : input_arg_expansions_) { in GrapplerFunctionItem()
350 const std::vector<InputArgExpansion>& GrapplerFunctionItem::inputs() const { in inputs()
354 const InputArgExpansion& GrapplerFunctionItem::input(int i) const { in input()
539 std::vector<InputArgExpansion> inputs; in MakeGrapplerFunctionItem()
554 InputArgExpansion input_expansion{/*input_name=*/input.name(), in MakeGrapplerFunctionItem()
669 for (const InputArgExpansion& input : item.inputs()) { in RegisterGrapplerFunctionConnectivity()
690 InputArgExpansion* input_arg_expansion = nullptr; in ReplaceInputWithConst()
[all …]
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dfunction_optimizer.cc977 for (const InputArgExpansion& input_arg : item.inputs()) { in InlinedFunctionInputsNode()
1033 for (const InputArgExpansion& input_arg : item.inputs()) { in InlineDirectFunctionCall()
1636 for (const InputArgExpansion& input_arg : item.inputs()) { in InlineIndirectFunctionCall()