Home
last modified time | relevance | path

Searched refs:model_output (Results 1 – 24 of 24) sorted by relevance

/external/tensorflow/tensorflow/lite/tools/signature/
Dsignature_def_util_test.cc68 std::string model_output; in TEST_F() local
73 SetSignatureDefMap(model_, expected_signature_def_map, &model_output)); in TEST_F()
74 const Model* add_model = flatbuffers::GetRoot<Model>(model_output.data()); in TEST_F()
87 std::string model_output; in TEST_F() local
92 SetSignatureDefMap(model_, expected_signature_def_map, &model_output)); in TEST_F()
93 const Model* add_model = flatbuffers::GetRoot<Model>(model_output.data()); in TEST_F()
109 SetSignatureDefMap(add_model, expected_signature_def_map, &model_output)); in TEST_F()
110 const Model* final_model = flatbuffers::GetRoot<Model>(model_output.data()); in TEST_F()
134 std::string model_output; in TEST_F() local
139 SetSignatureDefMap(model_, expected_signature_def_map, &model_output)); in TEST_F()
[all …]
/external/armnn/python/pyarmnn/examples/keyword_spotting/
Daudio_utils.py9 def decode(model_output: np.ndarray, labels: dict) -> list:
19 results = [labels[np.argmax(model_output)], model_output[0][0][np.argmax(model_output)]]
/external/executorch/backends/xnnpack/test/
Dtest_xnnpack_utils.py136 def assert_outputs_equal(self, model_output, ref_output): argument
147 self.assertTrue(len(ref_output) == len(model_output))
151 model_output[i], ref_output[i], atol=1e-03, rtol=1e-03
157 torch.allclose(model_output[0], ref_output, atol=1e-03, rtol=1e-03)
246 model_output = executorch_module.run_method("forward", tuple(inputs_flattened))
248 self.assert_outputs_equal(model_output, ref_output)
458 model_output = executorch_module.run_method("forward", tuple(inputs_flattened))
462 print("Model_output:", model_output[0])
466 torch.allclose(model_output[0], ref_output, atol=4e-03, rtol=1e-03)
/external/executorch/exir/backend/test/demos/rpc/
Dtest_rpc.py135 model_output = executorch_module.run_method("forward", tuple(inputs_flattened))
140 torch.allclose(model_output[0], ref_output, rtol=1e-03, atol=1e-03)
181 model_output = executorch_module.run_method("forward", tuple(inputs_flattened))
186 torch.allclose(model_output[0], ref_output, rtol=1e-03, atol=1e-03)
/external/executorch/backends/apple/mps/test/
Dtest_mps_utils.py127 model_output, argument
143 model_output
146 res_output = model_output[i].cpu()
167 res_output = model_output[0].cpu()
279 model_output = executorch_module.forward(original_inputs)
283 self.assert_outputs_equal(model_output, expected_output, atol, rtol)
/external/libopus/dnn/torch/rdovae/
Dtrain_rdovae.py194 model_output = model(features, q_ids) variable
197 z = model_output['z']
198 states = model_output['states']
199 outputs_hard_quant = model_output['outputs_hard_quant']
200 outputs_soft_quant = model_output['outputs_soft_quant']
201 statistical_model = model_output['statistical_model']
/external/armnn/python/pyarmnn/examples/speech_recognition/
Daudio_utils.py9 def decode(model_output: np.ndarray, labels: dict) -> str:
19 top1_results = [labels[np.argmax(row)] for row in model_output]
/external/executorch/exir/backend/test/demos/
Dtest_delegate_aten_mode.py65 model_output = executorch_module.run_method("forward", tuple(inputs_flattened))
69 torch.allclose(model_output[0], ref_output, atol=1e-03, rtol=1e-03)
Dtest_xnnpack_qnnpack.py128 model_output = executorch_module.run_method("forward", tuple(inputs_flattened))
133 torch.allclose(model_output[0], ref_output, atol=1e-03, rtol=1e-03)
/external/federated-compute/fcp/artifact_building/
Dplan_utils_test.py108 model_output = interpreter.get_output_details()[0]
113 self.assertEqual(interpreter.get_tensor(model_output['index']), 9)
182 model_output = interpreter.get_output_details()[0]
187 self.assertEqual(interpreter.get_tensor(model_output['index']), 9)
/external/executorch/backends/apple/coreml/test/
Dtest_coreml_quantizer.py60 model_output = model(*example_inputs).detach().numpy()
62 np.testing.assert_allclose(quantized_output, model_output, rtol=5e-2, atol=5e-2)
/external/executorch/backends/qualcomm/tests/
Dutils.py135 def _assert_outputs_equal(self, model_output, ref_output): argument
136 self.assertTrue(len(ref_output) == len(model_output))
140 model_output[i], ref_output[i], atol=self.atol, rtol=self.rtol
/external/executorch/backends/vulkan/test/
Dtest_vulkan_delegate.py42 model_output, argument
59 self.assertTrue(len(ref_output) == len(model_output))
63 model_output[0],
74 model_output[i],
85 model_output[0],
142 model_output = executorch_module.run_method(
148 model_output,
158 model_output = executorch_module.run_method(
164 model_output,
/external/executorch/examples/qualcomm/qaihub_scripts/stable_diffusion/runner/
Drunner.h86 const std::vector<float>& model_output,
Drunner.cpp329 const std::vector<float>& model_output, in step() argument
338 float sigma_hat = sample[i] - (sigma * model_output[i]); in step()
/external/executorch/backends/xnnpack/test/tester/
Dtester.py667 def _assert_outputs_equal(model_output, ref_output, atol=1e-03, rtol=1e-03): argument
677 assert len(model_output) == len(ref_output)
679 for i in range(len(model_output)):
680 model = model_output[i]
/external/googleapis/google/cloud/dialogflow/cx/v3beta1/
Ddata_store_connection.proto71 string model_output = 2; field
92 string model_output = 2; field
/external/executorch/backends/apple/coreml/runtime/delegate/
DETCoreMLModelManager.mm115 MLMultiArray *model_output = [enumerator nextObject];
116 ::copy(model_output, output);
155 MLMultiArray *model_output = [enumerator nextObject];
156 ::copy(model_output, output);
/external/executorch/exir/backend/test/
Dtest_backends_lifted.py283 model_output = executorch_module.run_method("forward", tuple(inputs_flattened))
287 torch.allclose(model_output[0], ref_output, atol=1e-03, rtol=1e-03)
868 model_output = executorch_module.run_method("forward", tuple(inputs_flattened))
872 torch.allclose(model_output[0], ref_output, atol=1e-03, rtol=1e-03),
Dtest_backends.py267 model_output = executorch_module.run_method("forward", tuple(inputs_flattened))
271 torch.allclose(model_output[0], ref_output, atol=1e-03, rtol=1e-03)
877 model_output = executorch_module.run_method("forward", tuple(inputs_flattened))
881 torch.allclose(model_output[0], ref_output, atol=1e-03, rtol=1e-03),
/external/executorch/exir/emit/test/
Dtest_emit.py1267 model_output = executorch_module.run_method("get_tensor_list", [])
1269 torch.allclose(model_output[i], tensor_list_output[i])
/external/tensorflow/tensorflow/python/framework/
Dextension_type_test.py1084 model_output = array_ops.identity(model_input, name='output')
1085 model = training.Model(inputs=model_input, outputs=model_output)
/external/pytorch/torch/testing/_internal/distributed/
Ddistributed_test.py9302 def get_loss(model_output): argument
9304 if isinstance(model_output, torch.Tensor):
9305 return model_output.sum()
9306 elif isinstance(model_output, dict):
9307 for value in model_output.values():
9309 elif isinstance(model_output, (tuple, list)):
9310 for x in model_output:
/external/pytorch/test/
Dtest_nn.py402 model_output = net(torch.randn([5, 10]))
403 model_output["output"].backward()