Searched refs:TensorsMap (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/lite/toco/tflite/ |
| D | export.cc | 172 void LoadTensorsMap(const Model& model, TensorsMap* tensors_map) { in LoadTensorsMap()
208 const Model& model, const details::TensorsMap& tensors_map, in ExportTensors()
268 const Model& model, const details::TensorsMap& tensors_map, in ExportInputTensors()
278 const Model& model, const details::TensorsMap& tensors_map, in ExportOutputTensors()
334 const details::TensorsMap& tensors_map, FlatBufferBuilder* builder, in ExportOperators()
440 details::TensorsMap tensors_map; in Export()
|
| D | export.h | 80 using TensorsMap = std::unordered_map<string, int>; variable
164 void LoadTensorsMap(const Model& model, TensorsMap* tensors_map);
|
| D | export_test.cc | 194 details::TensorsMap tensors; in TEST_F()
|