Searched refs:input_nodes_ (Results 1 – 2 of 2) sorted by relevance
129 std::unordered_map<Output, int, OutputHash, OutputEq> input_nodes_; member in tensorflow::__anon2882cf250111::SymbolicGradientBuilder276 input_nodes_.reserve(inputs_.size()); in Initialize()278 input_nodes_.insert({inputs_[i], i}); in Initialize()458 auto iter = input_nodes_.find({n, i}); in AddGradients()459 if (iter != input_nodes_.end()) { in AddGradients()542 int idx = input_nodes_[nout]; in AddGradients()
101 std::vector<string> input_nodes_; member in tensorflow::tensorrt::TRTEngineOp