Searched refs:GetInputTensor (Results 1 – 14 of 14) sorted by relevance
135 const TfLiteTensor* GetInputTensor(int index) { in GetInputTensor() function in tflite::benchmark::__anone243b2400211::TestBenchmark260 auto input_tensor = benchmark.GetInputTensor(0); in TEST()290 CheckInputTensorValue(benchmark.GetInputTensor(1), file_value_b); in TEST()291 CheckInputTensorValue(benchmark.GetInputTensor(3), file_value_d); in TEST()310 CheckInputTensorValue(benchmark.GetInputTensor(0), file_value); in TEST()337 auto input_tensor = benchmark.GetInputTensor(0); in TEST()
120 const TfLiteTensor* input0 = reader->GetInputTensor(0); in ParseInputsWithConstTensor()125 const TfLiteTensor* input1 = reader->GetInputTensor(1); in ParseInputsWithConstTensor()296 const TfLiteTensor* second_input = reader->GetInputTensor(1); in Parse()592 const TfLiteTensor* src_tensor = reader->GetInputTensor(0); in Parse()593 const TfLiteTensor* weights_tensor = reader->GetInputTensor(1); in Parse()637 const TfLiteTensor* src_tensor = reader->GetInputTensor(0); in ReadAttributes()645 const TfLiteTensor* weights_tensor = reader->GetInputTensor(1); in ReadAttributes()671 const TfLiteTensor* src_tensor = reader->GetInputTensor(0); in ResolveGroupedConvolution()780 const TfLiteTensor* input_tensor = reader->GetInputTensor(0); in Parse()781 const TfLiteTensor* axis_tensor = reader->GetInputTensor(1); 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()
116 ConstTensorViewOr GetInputTensor(const int idx) const { in GetInputTensor() function117 return static_cast<const SubType&>(*this).GetInputTensor(idx); in GetInputTensor()
79 ConstTensorViewOr GetInputTensor(const int idx) const;
141 ConstTensorViewOr TfShapeInferenceContext::GetInputTensor(const int idx) const { in GetInputTensor() function in tflite::shim::TfShapeInferenceContext
87 ConstTensorViewOr GetInputTensor(const int idx) const;
152 ConstTensorViewOr TfLiteShapeInferenceContext::GetInputTensor( in GetInputTensor() function in tflite::shim::TfLiteShapeInferenceContext
169 ConstTensorViewOr GetInputTensor(const int idx) const;
141 const auto input_t_or = ctx->GetInputTensor(kInput0); in ShapeInference()
418 py::object GetInputTensor(std::string& input_name, const py::dict& keywords, in GetInputTensor() function489 py::object input_tensor = GetInputTensor(input_name, keywords, op_def); in ExtractInputsAndAttrs()
1392 const armnn::ConstTensor GetInputTensor(const LayerBindingId layerId, const InputTensors& inputTens… in GetInputTensor() function