Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/grappler/optimizers/
Dconstant_folding.h68 bool ForwardInputs(NodeDef* node, absl::Span<const int> inputs_to_forward);
Dconstant_folding.cc259 absl::Span<const int> inputs_to_forward) { in ForwardInputs() argument
260 for (int input_idx : inputs_to_forward) { in ForwardInputs()
269 for (int input_idx : inputs_to_forward) { in ForwardInputs()
911 std::vector<int> inputs_to_forward; in MaterializeOutputValues() local
912 std::iota(inputs_to_forward.begin(), inputs_to_forward.end(), 0); in MaterializeOutputValues()
913 graph_modified_ = ForwardInputs(node, inputs_to_forward); in MaterializeOutputValues()
3565 std::vector<int> inputs_to_forward; in PartialConstPropThroughIdentityN() local
3579 inputs_to_forward.push_back(input_idx); in PartialConstPropThroughIdentityN()
3582 return ForwardInputs(node, inputs_to_forward); in PartialConstPropThroughIdentityN()