Searched refs:GetInputTensor (Results 1 – 12 of 12) sorted by relevance
132 const TfLiteTensor* GetInputTensor(int index) { in GetInputTensor() function in tflite::benchmark::__anona75ccbf60211::TestBenchmark233 auto input_tensor = benchmark.GetInputTensor(0); in TEST()263 CheckInputTensorValue(benchmark.GetInputTensor(1), file_value_b); in TEST()264 CheckInputTensorValue(benchmark.GetInputTensor(3), file_value_d); in TEST()283 CheckInputTensorValue(benchmark.GetInputTensor(0), file_value); in TEST()310 auto input_tensor = benchmark.GetInputTensor(0); in TEST()
116 ConstTensorViewOr GetInputTensor(const int idx) const { in GetInputTensor() function117 return static_cast<const SubType&>(*this).GetInputTensor(idx); in GetInputTensor()
78 ConstTensorViewOr GetInputTensor(const int idx) const;
87 ConstTensorViewOr GetInputTensor(const int idx) const;
148 ConstTensorViewOr TfLiteShapeInferenceContext::GetInputTensor( in GetInputTensor() function in tflite::shim::TfLiteShapeInferenceContext
139 ConstTensorViewOr TfShapeInferenceContext::GetInputTensor(const int idx) const { in GetInputTensor() function in tflite::shim::TfShapeInferenceContext
169 ConstTensorViewOr GetInputTensor(const int idx) const;
118 const TfLiteTensor* input0 = reader->GetInputTensor(0); in ParseInputsWithConstTensor()123 const TfLiteTensor* input1 = reader->GetInputTensor(1); in ParseInputsWithConstTensor()537 const TfLiteTensor* const_tensor = reader->GetInputTensor(0); in Parse()585 const TfLiteTensor* input = reader->GetInputTensor(0); in Parse()586 const TfLiteTensor* filter = reader->GetInputTensor(1); in Parse()677 const TfLiteTensor* tflite_input = reader->GetInputTensor(0); in Parse()792 IsConstantTensor(reader->GetInputTensor(0)); in Parse()1148 const TfLiteTensor* input0 = reader->GetInputTensor(0); in Parse()1153 const TfLiteTensor* input1 = reader->GetInputTensor(1); in Parse()1444 const TfLiteTensor* input = reader->GetInputTensor(0); in Parse()[all …]
131 TfLiteTensor* GetInputTensor(int index) const;
224 TfLiteTensor* ObjectReader::GetInputTensor(int index) const { in GetInputTensor() function in tflite::gpu::ObjectReader
58 reader->GetInputTensor(weights_tensor_id); in GetFullyConnectedNode()
134 const auto input_t_or = ctx->GetInputTensor(kInput0); in ShapeInference()