Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/cc/framework/
Dgradients.cc130 std::unordered_map<Output, int, OutputHash, OutputEq> input_nodes_; member in tensorflow::__anonc8091ec30111::SymbolicGradientBuilder
277 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()
/external/tensorflow/tensorflow/core/grappler/utils/
Dfunctions.cc429 return input_nodes_.contains(node.name()); in IsInputNode()
437 absl::flat_hash_set<absl::string_view> input_nodes_; member in tensorflow::grappler::__anoneaf86e2d0511::MakeFunctionDefHelper
446 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()
/external/tensorflow/tensorflow/compiler/tf2tensorrt/kernels/
Dtrt_engine_op.cc152 std::vector<string> input_nodes_; member in tensorflow::tensorrt::TRTEngineOp