Searched refs:InputTensor (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/core/graph/ |
| D | graph.h | 315 struct InputTensor { struct
319 InputTensor(Node* n, int i) : node(n), index(i) {} in InputTensor() argument
320 InputTensor() : node(nullptr), index(0) {} in InputTensor() function
324 bool operator==(const InputTensor& other) const;
329 uint64 operator()(InputTensor const& s) const; argument
|
| D | graph.cc | 327 bool InputTensor::operator==(const InputTensor& other) const { in operator ==()
331 uint64 InputTensor::Hash::operator()(InputTensor const& s) const { in operator ()()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | encapsulate_subgraphs_pass.cc | 130 uint64 operator()(std::pair<OutputTensor, InputTensor> const& s) const { in operator ()()
132 InputTensor::Hash()(s.second)); in operator ()()
393 std::unordered_map<InputTensor, int, InputTensor::Hash> outputs_by_dst;
480 std::unordered_map<InputTensor, int, InputTensor::Hash> args_by_dst_;
583 std::unordered_set<std::pair<OutputTensor, InputTensor>,
716 return args_by_dst_.at(InputTensor(edge->dst(), edge->dst_input())); in GetArgIndexForEdge()
744 .outputs_by_dst.at(InputTensor(edge->dst(), edge->dst_input())); in GetSendFromHostSlot()
800 args_by_dst_[InputTensor(dst_node, dst_slot)] = arg_index; in RecordArg()
874 ->outputs_by_dst[InputTensor(edge->dst(), edge->dst_input())] = in RecordOutsideCompilationOutputOrControl()
1722 std::unordered_set<std::pair<OutputTensor, InputTensor>, in CopyEdgeToOutputGraph() argument
[all …]
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | lstm.cc | 645 enum InputTensor { enum
|