Searched refs:InputTensor (Results 1 – 6 of 6) 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()
914 std::unordered_set<std::pair<OutputTensor, InputTensor>, in CopyEdgeToOutputGraph() argument
929 ->emplace(OutputTensor(src_image, -1), InputTensor(dst_image, -1)) in CopyEdgeToOutputGraph()
945 InputTensor(dst_image, dst_input)) in CopyEdgeToOutputGraph()
958 std::unordered_set<std::pair<OutputTensor, InputTensor>, in AddEdgesToOutputGraph()
|
| /external/tensorflow/tensorflow/core/graph/ |
| D | graph.h | 364 struct InputTensor { struct
368 InputTensor(Node* n, int i) : node(n), index(i) {} in InputTensor() argument
369 InputTensor() : node(nullptr), index(0) {} in InputTensor() argument
373 bool operator==(const InputTensor& other) const;
378 uint64 operator()(InputTensor const& s) const; argument
|
| D | graph.cc | 339 bool InputTensor::operator==(const InputTensor& other) const { in operator ==()
343 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/parse_example/ |
| D | parse_example.cc | 644 enum InputTensor { enum
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | lstm.cc | 2165 enum InputTensor { enum
|