| /aosp_15_r20/external/tensorflow/tensorflow/compiler/jit/ |
| H A D | encapsulate_subgraphs_pass.cc | 118 uint64 operator()(std::pair<OutputTensor, InputTensor> const& s) const { in operator ()()
120 InputTensor::Hash()(s.second)); in operator ()()
264 std::unordered_map<InputTensor, int, InputTensor::Hash> args_by_dst_;
339 std::unordered_set<std::pair<OutputTensor, InputTensor>,
419 return args_by_dst_.at(InputTensor(edge->dst(), edge->dst_input())); in GetArgIndexForEdge()
477 args_by_dst_[InputTensor(dst_node, dst_slot)] = arg_index; in RecordArg()
888 std::unordered_set<std::pair<OutputTensor, InputTensor>, in CopyEdgeToOutputGraph() argument
903 ->emplace(OutputTensor(src_image, -1), InputTensor(dst_image, -1)) in CopyEdgeToOutputGraph()
919 InputTensor(dst_image, dst_input)) in CopyEdgeToOutputGraph()
932 std::unordered_set<std::pair<OutputTensor, InputTensor>, in AddEdgesToOutputGraph()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/graph/ |
| H A D | graph.h | 384 struct InputTensor { struct
388 InputTensor(Node* n, int i) : node(n), index(i) {} in InputTensor() argument
389 InputTensor() : node(nullptr), index(0) {} in InputTensor() argument
393 bool operator==(const InputTensor& other) const;
398 uint64 operator()(InputTensor const& s) const; argument
|
| H A D | graph.cc | 364 bool InputTensor::operator==(const InputTensor& other) const { in operator ==()
368 uint64 InputTensor::Hash::operator()(InputTensor const& s) const { in operator ()()
|
| /aosp_15_r20/external/armnn/src/armnnTfLiteParser/test/ |
| H A D | LocalResponseNormalization.cpp | 45 "name": "InputTensor", in LRNFixture()
|
| H A D | Constant.cpp | 42 "name": "InputTensor", in ConstantAddFixture()
|
| H A D | ResizeNearestNeighbor.cpp | 40 "name": "InputTensor", in ResizeNearestNeighborFixture()
|
| H A D | ResizeBilinear.cpp | 40 "name": "InputTensor", in ResizeBilinearFixture()
|
| H A D | AvgPool2D.cpp | 44 "name": "InputTensor", in AvgPool2DFixture()
|
| H A D | MaxPool2D.cpp | 45 "name": "InputTensor", in MaxPool2DFixture()
|
| H A D | GetTensorIds.cpp | 40 "name": "InputTensor", in GetTensorIdsFixture()
|
| H A D | GetSubgraphInputsOutputs.cpp | 41 "name": "InputTensor", in GetSubgraphInputsOutputsMainFixture()
|
| H A D | GetInputsOutputs.cpp | 40 "name": "InputTensor", in GetInputsOutputsMainFixture()
|
| H A D | LoadModel.cpp | 44 "name": "InputTensor", in LoadModelFixture()
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/ |
| H A D | unidirectional_sequence_gru.cc | 89 enum InputTensor { enum
|
| H A D | lstm.cc | 2138 enum InputTensor { enum
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/parse_example/ |
| H A D | parse_example.cc | 645 enum InputTensor { enum
|