Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/jit/
Dencapsulate_subgraphs_pass.cc282 std::unordered_map<OutputTensor, int, OutputTensor::Hash> args_by_src_; member in tensorflow::__anona26b85820211::Encapsulator::Subgraph
475 std::tie(iter, inserted) = args_by_src_.emplace( in RecordArg()
476 OutputTensor(src_node, src_slot), args_by_src_.size()); in RecordArg()
578 std::vector<OutputTensor> arg_source_tensors(args_by_src_.size()); in BuildFunctionDef()
579 for (const auto& arg : args_by_src_) { in BuildFunctionDef()
583 std::vector<int> input_permutation(args_by_src_.size()); in BuildFunctionDef()
595 if (input_permutation.size() != args_by_src_.size()) { in BuildFunctionDef()
601 for (auto& arg : args_by_src_) { in BuildFunctionDef()