Searched refs:input_nodes_ (Results 1 – 3 of 3) sorted by relevance
130 std::unordered_map<Output, int, OutputHash, OutputEq> input_nodes_; member in tensorflow::__anonc8091ec30111::SymbolicGradientBuilder277 input_nodes_.reserve(inputs_.size()); in Initialize()279 input_nodes_.insert({inputs_[i], i}); in Initialize()459 auto iter = input_nodes_.find({n, i}); in AddGradients()460 if (iter != input_nodes_.end()) { in AddGradients()543 int idx = input_nodes_[nout]; in AddGradients()
429 return input_nodes_.contains(node.name()); in IsInputNode()437 absl::flat_hash_set<absl::string_view> input_nodes_; member in tensorflow::grappler::__anoneaf86e2d0511::MakeFunctionDefHelper446 input_nodes_.insert(input_arg.node_name); in Initialize()477 const auto is_input = input_nodes_.find(tensor.node()); in AsFunctionDefInput()478 if (is_input != input_nodes_.end()) { in AsFunctionDefInput()
152 std::vector<string> input_nodes_; member in tensorflow::tensorrt::TRTEngineOp