Searched refs:InputTensor (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | encapsulate_subgraphs_pass.cc | 134 uint64 operator()(std::pair<OutputTensor, InputTensor> const& s) const { in operator ()()
136 InputTensor::Hash()(s.second)); in operator ()()
283 std::unordered_map<InputTensor, int, InputTensor::Hash> args_by_dst_;
358 std::unordered_set<std::pair<OutputTensor, InputTensor>,
438 return args_by_dst_.at(InputTensor(edge->dst(), edge->dst_input())); in GetArgIndexForEdge()
498 args_by_dst_[InputTensor(dst_node, dst_slot)] = arg_index; in RecordArg()
917 std::unordered_set<std::pair<OutputTensor, InputTensor>, in CopyEdgeToOutputGraph() argument
932 ->emplace(OutputTensor(src_image, -1), InputTensor(dst_image, -1)) in CopyEdgeToOutputGraph()
948 InputTensor(dst_image, dst_input)) in CopyEdgeToOutputGraph()
961 std::unordered_set<std::pair<OutputTensor, InputTensor>, in AddEdgesToOutputGraph()
|
| /external/tensorflow/tensorflow/core/graph/ |
| D | graph.h | 341 struct InputTensor { struct
345 InputTensor(Node* n, int i) : node(n), index(i) {} in InputTensor() argument
346 InputTensor() : node(nullptr), index(0) {} in InputTensor() argument
350 bool operator==(const InputTensor& other) const;
355 uint64 operator()(InputTensor const& s) const; argument
|
| D | graph.cc | 349 bool InputTensor::operator==(const InputTensor& other) const { in operator ==()
353 uint64 InputTensor::Hash::operator()(InputTensor const& s) const { in operator ()()
|
| /external/tensorflow/tensorflow/lite/experimental/kernels/ |
| D | unidirectional_sequence_gru.cc | 81 enum InputTensor { enum
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | lstm.cc | 1212 enum InputTensor { enum
|