Home
last modified time | relevance | path

Searched refs:InputLayerInfo (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/tools/benchmark/
Dbenchmark_model.h26 struct InputLayerInfo { struct
39 Status RunBenchmark(const std::vector<InputLayerInfo>& inputs, argument
46 const std::vector<InputLayerInfo>& inputs,
Dbenchmark_model_test.cc30 benchmark_model::InputLayerInfo* input, in CreateTestGraph()
55 benchmark_model::InputLayerInfo input; in TEST()
83 benchmark_model::InputLayerInfo input; in TEST()
Dbenchmark_model.cc76 const std::vector<InputLayerInfo>& inputs, in CreateTensorsFromInputInfo()
78 for (const InputLayerInfo& input : inputs) { in CreateTensorsFromInputInfo()
117 Status GetOutputShapes(const std::vector<InputLayerInfo>& inputs, in GetOutputShapes()
150 const std::vector<InputLayerInfo>& inputs, in CalculateFlops()
284 Status RunBenchmark(const std::vector<InputLayerInfo>& inputs, in RunBenchmark()
322 const std::vector<InputLayerInfo>& inputs, in TimeMultipleRuns()
531 std::vector<InputLayerInfo> inputs; in Main()
533 InputLayerInfo input; in Main()
/external/tensorflow/tensorflow/lite/tools/benchmark/
Dbenchmark_tflite_model.h36 struct InputLayerInfo { struct
37 InputLayerInfo() : has_value_range(false) {} in InputLayerInfo() function
116 const InputLayerInfo* layer_info);
121 std::vector<InputLayerInfo> inputs_;
Dbenchmark_tflite_model.cc108 int FindLayerInfoIndex(std::vector<BenchmarkTfLiteModel::InputLayerInfo>* info, in FindLayerInfoIndex()
123 std::vector<BenchmarkTfLiteModel::InputLayerInfo>* info) { in PopulateInputValueRanges()
154 std::vector<BenchmarkTfLiteModel::InputLayerInfo>* info) { in PopulateInputValueFiles()
181 std::vector<BenchmarkTfLiteModel::InputLayerInfo>* info) { in PopulateInputLayerInfo()
199 info->push_back(BenchmarkTfLiteModel::InputLayerInfo()); in PopulateInputLayerInfo()
200 BenchmarkTfLiteModel::InputLayerInfo& input = info->back(); in PopulateInputLayerInfo()
435 const InputLayerInfo* layer_info) { in CreateRandomTensorData()
544 const InputLayerInfo* input_layer_info = nullptr; in PrepareInputData()
700 const InputLayerInfo& input = inputs_[j]; in Init()
717 const InputLayerInfo& input = inputs_[j]; in Init()