Home
last modified time | relevance | path

Searched refs:SetInputTensor (Results 1 – 6 of 6) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/metal/benchmarking/
H A Dmain.mm170 RETURN_IF_ERROR(inference_context.SetInputTensor(input->id, src_tensor));
299 RETURN_IF_ERROR(inference_context.SetInputTensor(graph->inputs()[i]->id, src_tensors[i]));
326 RETURN_IF_ERROR(serialized_context.SetInputTensor(graph->inputs()[i]->id, src_tensors[i]));
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/cl/testing/
H A Dperformance_profiling.cc172 RETURN_IF_ERROR(context.SetInputTensor(graph_cl.inputs()[i]->id, in RunSerializedTest()
197 RETURN_IF_ERROR(serialized_context.SetInputTensor( in RunSerializedTest()
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/metal/
H A Dinference_context.h132 absl::Status SetInputTensor(ValueId id, const TensorFloat32& tensor);
H A Dinference_context.cc345 absl::Status InferenceContext::SetInputTensor(ValueId id, in SetInputTensor() function in tflite::gpu::metal::InferenceContext
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/cl/
H A Dinference_context.h92 absl::Status SetInputTensor(ValueId id, const TensorFloat32& tensor,
H A Dinference_context.cc902 absl::Status InferenceContext::SetInputTensor(ValueId id, in SetInputTensor() function in tflite::gpu::cl::InferenceContext