Searched refs:InputTensor (Results 1 – 15 of 15) 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 ()()
280 std::unordered_map<InputTensor, int, InputTensor::Hash> args_by_dst_;
355 std::unordered_set<std::pair<OutputTensor, InputTensor>,
435 return args_by_dst_.at(InputTensor(edge->dst(), edge->dst_input())); in GetArgIndexForEdge()
495 args_by_dst_[InputTensor(dst_node, dst_slot)] = arg_index; in RecordArg()
911 std::unordered_set<std::pair<OutputTensor, InputTensor>, in CopyEdgeToOutputGraph() argument
926 ->emplace(OutputTensor(src_image, -1), InputTensor(dst_image, -1)) in CopyEdgeToOutputGraph()
942 InputTensor(dst_image, dst_input)) in CopyEdgeToOutputGraph()
955 std::unordered_set<std::pair<OutputTensor, InputTensor>, in AddEdgesToOutputGraph()
|
| /external/tensorflow/tensorflow/core/graph/ |
| D | graph.h | 374 struct InputTensor { struct
378 InputTensor(Node* n, int i) : node(n), index(i) {} in InputTensor() argument
379 InputTensor() : node(nullptr), index(0) {} in InputTensor() function
383 bool operator==(const InputTensor& other) const;
388 uint64 operator()(InputTensor const& s) const; argument
|
| D | graph.cc | 344 bool InputTensor::operator==(const InputTensor& other) const { in operator ==()
348 uint64 InputTensor::Hash::operator()(InputTensor const& s) const { in operator ()()
|
| /external/armnn/src/armnnTfLiteParser/test/ |
| D | Constant.cpp | 46 "name": "InputTensor", in ConstantAddFixture()
|
| D | ResizeBilinear.cpp | 46 "name": "InputTensor", in ResizeBilinearFixture()
|
| D | ResizeNearestNeighbor.cpp | 46 "name": "InputTensor", in ResizeNearestNeighborFixture()
|
| D | AvgPool2D.cpp | 46 "name": "InputTensor", in AvgPool2DFixture()
|
| D | MaxPool2D.cpp | 46 "name": "InputTensor", in MaxPool2DFixture()
|
| D | GetTensorIds.cpp | 41 "name": "InputTensor", in GetTensorIdsFixture()
|
| D | GetSubgraphInputsOutputs.cpp | 42 "name": "InputTensor", in GetSubgraphInputsOutputsMainFixture()
|
| D | GetInputsOutputs.cpp | 41 "name": "InputTensor", in GetInputsOutputsMainFixture()
|
| D | LoadModel.cpp | 45 "name": "InputTensor", in LoadModelFixture()
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | unidirectional_sequence_gru.cc | 89 enum InputTensor { enum
|
| D | lstm.cc | 2134 enum InputTensor { enum
|
| /external/tensorflow/tensorflow/lite/kernels/parse_example/ |
| D | parse_example.cc | 644 enum InputTensor { enum
|