Home
last modified time | relevance | path

Searched refs:inference_params (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/lite/tools/evaluation/tasks/inference_diff/
Drun_eval.cc83 auto* inference_params = in RunImpl() local
85 inference_params->set_model_file_path(model_file_path_); in RunImpl()
86 inference_params->set_num_threads(num_interpreter_threads_); in RunImpl()
89 inference_params->set_invocations_per_run(3); in RunImpl()
90 inference_params->set_delegate(ParseStringToDelegateType(delegate_)); in RunImpl()
92 inference_params->delegate() == TfliteInferenceParams::NONE) { in RunImpl()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dcudnn_batchnorm_runner.cc231 CudnnBatchNormForwardInferenceParams inference_params; in RunCudnnBatchNormForwardInference() local
232 AssignCommonParams(config, &inference_params.common, operand, scale); in RunCudnnBatchNormForwardInference()
233 inference_params.offset = offset; in RunCudnnBatchNormForwardInference()
234 inference_params.mean = mean; in RunCudnnBatchNormForwardInference()
235 inference_params.variance = variance; in RunCudnnBatchNormForwardInference()
236 inference_params.output = output; in RunCudnnBatchNormForwardInference()
240 RunCudnnBatchNormForwardInferenceImpl<Eigen::half>(&inference_params, in RunCudnnBatchNormForwardInference()
244 RunCudnnBatchNormForwardInferenceImpl<float>(&inference_params, stream); in RunCudnnBatchNormForwardInference()
/external/tensorflow/tensorflow/lite/tools/evaluation/tasks/imagenet_image_classification/
Drun_eval.cc152 auto* inference_params = classification_params->mutable_inference_params(); in RunImpl() local
153 inference_params->set_model_file_path(model_file_path_); in RunImpl()
154 inference_params->set_num_threads(num_interpreter_threads_); in RunImpl()
155 inference_params->set_delegate(ParseStringToDelegateType(delegate_)); in RunImpl()
/external/tensorflow/tensorflow/lite/tools/evaluation/tasks/coco_object_detection/
Drun_eval.cc129 auto* inference_params = detection_params->mutable_inference_params(); in RunImpl() local
130 inference_params->set_model_file_path(model_file_path_); in RunImpl()
131 inference_params->set_num_threads(num_interpreter_threads_); in RunImpl()
132 inference_params->set_delegate(ParseStringToDelegateType(delegate_)); in RunImpl()
/external/tensorflow/tensorflow/lite/tools/evaluation/stages/
Dobject_detection_stage.cc49 ->mutable_tflite_inference_params() = params.inference_params(); in Init()
Dimage_classification_stage.cc49 ->mutable_tflite_inference_params() = params.inference_params(); in Init()
/external/tensorflow/tensorflow/lite/tools/evaluation/proto/
Devaluation_stages.proto164 optional TfliteInferenceParams inference_params = 1; field
298 optional TfliteInferenceParams inference_params = 1; field