Searched refs:inferenceInput (Results 1 – 3 of 3) sorted by relevance
66 InferenceInput inferenceInput = in buildInput_success() local74 float[] inputData = (float[]) inferenceInput.getInputData()[0]; in buildInput_success()76 assertThat(inferenceInput.getBatchSize()).isEqualTo(1); in buildInput_success()77 assertThat(inferenceInput.getExpectedOutputStructure().getDataOutputs()).hasSize(1); in buildInput_success()78 assertThat(inferenceInput.getParams()).isEqualTo(params); in buildInput_success()90 InferenceInput inferenceInput = in buildInputWithSetters() local102 float[] inputData = (float[]) inferenceInput.getInputData()[0]; in buildInputWithSetters()104 assertThat(inferenceInput.getBatchSize()).isEqualTo(1); in buildInputWithSetters()105 assertThat(inferenceInput.getExpectedOutputStructure().getDataOutputs()).hasSize(1); in buildInputWithSetters()106 assertThat(inferenceInput.getParams()).isEqualTo(params); in buildInputWithSetters()[all …]
82 InferenceInput inferenceInput = in runModelInference_singleExample_success() local90 Constants.EXTRA_INFERENCE_INPUT, new InferenceInputParcel(inferenceInput)); in runModelInference_singleExample_success()107 InferenceInput inferenceInput = in runModelInference_setBatchSizeNotMatch_success() local118 Constants.EXTRA_INFERENCE_INPUT, new InferenceInputParcel(inferenceInput)); in runModelInference_setBatchSizeNotMatch_success()134 InferenceInput inferenceInput = in runModelInferenceBatch_setBatchSize_success() local145 Constants.EXTRA_INFERENCE_INPUT, new InferenceInputParcel(inferenceInput)); in runModelInferenceBatch_setBatchSize_success()161 InferenceInput inferenceInput = in runModelInferenceBatch_notSetBatchSize_success() local171 Constants.EXTRA_INFERENCE_INPUT, new InferenceInputParcel(inferenceInput)); in runModelInferenceBatch_notSetBatchSize_success()191 InferenceInput inferenceInput = in runModelInference_invalidInputFormat() local198 Constants.EXTRA_INFERENCE_INPUT, new InferenceInputParcel(inferenceInput)); in runModelInference_invalidInputFormat()[all …]
77 InferenceInput inferenceInput = builder.setBatchSize(10).build(); in buildInferenceInput_reusable() local78 assertThat(inferenceInput.getBatchSize()).isEqualTo(10); in buildInferenceInput_reusable()90 InferenceInput inferenceInput = in buildInput_success() local98 float[] inputData = (float[]) inferenceInput.getInputData()[0]; in buildInput_success()100 assertThat(inferenceInput.getBatchSize()).isEqualTo(1); in buildInput_success()101 assertThat(inferenceInput.getExpectedOutputStructure().getDataOutputs()).hasSize(1); in buildInput_success()102 assertThat(inferenceInput.getParams()).isEqualTo(params); in buildInput_success()114 InferenceInput inferenceInput = in buildInput_batchNotSet_success() local121 assertThat(inferenceInput.getBatchSize()).isEqualTo(1); in buildInput_batchNotSet_success()