Home
last modified time | relevance | path

Searched refs:model_outputs (Results 1 – 20 of 20) sorted by relevance

/aosp_15_r20/external/pytorch/torch/onnx/_internal/
H A Dio_adapter.py98 model_outputs: Any,
119 model_outputs: Any,
132 model_outputs = step.apply(model_outputs, model=model)
133 return model_outputs
461 model_outputs: Any,
473 return pytree.tree_leaves(model_outputs)
487 model_outputs: Any,
503 for output in model_outputs
518 model_outputs: Any,
535 flattened_outputs, spec = pytree.tree_flatten(model_outputs)
[all …]
H A D_exporter_legacy.py684 model_outputs: Any,
738 …return self._output_adapter.apply(model_outputs, model=model_with_state_dict) # type: ignore[retu…
/aosp_15_r20/external/tensorflow/tensorflow/c/eager/
H A Dunified_api_testutil.cc69 std::vector<AbstractTensorHandle*> model_outputs; in RunModel() local
70 model_outputs.resize(outputs.size()); in RunModel()
72 absl::MakeSpan(model_outputs))); in RunModel()
80 for (int i = 0; i < model_outputs.size(); i++) { in RunModel()
81 if (model_outputs[i]) { in RunModel()
82 output_list.outputs.emplace_back(model_outputs[i]); in RunModel()
H A Dgradient_checker.cc53 AbstractTensorHandle* model_outputs[1]; in RunAndMaybeSum() local
57 RunModel(forward, ctx, inputs, model_outputs, use_function)); in RunAndMaybeSum()
58 AbstractTensorHandlePtr model_out(model_outputs[0]); in RunAndMaybeSum()
/aosp_15_r20/external/executorch/devtools/bundled_program/test/
H A Dtest_config.py115 model_outputs = getattr(
118 if isinstance(model_outputs, get_args(DataContainer)):
120 flatten_eager_model_outputs = tree_flatten(model_outputs)
123 model_outputs,
/aosp_15_r20/external/tensorflow/tensorflow/python/keras/engine/
H A Dfunctional.py1377 model_outputs = []
1384 model_outputs.append(
1386 model_outputs = nest.pack_sequence_as(network._nested_outputs, model_outputs)
1388 if not nest.is_nested(model_outputs):
1389 model_outputs = [model_outputs]
1390 model_outputs = tf_utils.convert_inner_node_data(model_outputs)
1391 config['output_layers'] = model_outputs
/aosp_15_r20/hardware/google/pixel/thermal/virtualtemp_estimator/
Dvirtualtemp_estimator_test.cpp361 std::vector<float> model_outputs; in run_batch_process() local
380 ret = vt_estimator_.Estimate(model_inputs, &model_outputs); in run_batch_process()
387 for (size_t i = 0; i < model_outputs.size(); ++i) { in run_batch_process()
388 model_outputs[i] /= kCelsius2mC; in run_batch_process()
389 model_out_string << model_outputs[i] << " "; in run_batch_process()
/aosp_15_r20/external/pytorch/torch/onnx/_internal/fx/
H A Ddynamo_graph_extractor.py122 model_outputs: Any,
127 return super().apply(model_outputs, model=model)
/aosp_15_r20/external/executorch/exir/backend/test/
H A Dtest_backends.py207 model_outputs = executorch_module.forward([model_inputs])
215 torch.allclose(model_outputs[0], expected_output, atol=1e-03, rtol=1e-03)
427 model_outputs = executorch_module.forward([model_inputs])
436 torch.allclose(model_outputs[0], expected_output, atol=1e-03, rtol=1e-03)
553 model_outputs = executorch_module.forward([*model_inputs])
564 model_outputs[index], expected_output, atol=1e-03, rtol=1e-03
H A Dtest_backends_lifted.py227 model_outputs = executorch_module.forward([model_inputs])
235 torch.allclose(model_outputs[0], expected_output, atol=1e-03, rtol=1e-03)
431 model_outputs = executorch_module.forward([model_inputs])
440 torch.allclose(model_outputs[0], expected_output, atol=1e-03, rtol=1e-03)
549 model_outputs = executorch_module.forward([*model_inputs])
560 model_outputs[index], expected_output, atol=1e-03, rtol=1e-03
H A Dtest_lowered_backend_module.py104 model_outputs = executorch_module.forward([model_inputs])
112 torch.allclose(model_outputs[0], expected_res, atol=1e-03, rtol=1e-03)
/aosp_15_r20/external/pytorch/torch/_inductor/
H A Dcompile_fx.py1212 model_outputs = model_outputs_node.args[0]
1214 n.name for n in model_outputs if isinstance(n, torch.fx.Node)
1370 model_outputs = pytree.arg_tree_leaves(*model_outputs_node.args)
1371 num_model_outputs = len(model_outputs)
1414 for n in model_outputs[original_output_start_index:orig_output_end_idx]
1457 model_outputs = pytree.arg_tree_leaves(*model_outputs_node.args)
1459 n.name for n in model_outputs if isinstance(n, torch.fx.Node)
/aosp_15_r20/hardware/google/pixel/thermal/
Dthermal-helper.cpp1080 std::vector<float> model_outputs; in runVirtualTempEstimator() local
1117 sensor_info.virtual_sensor_info->vt_estimator->Estimate(model_inputs, &model_outputs); in runVirtualTempEstimator()
1121 thermal_predictions_helper_.updateSensor(sensor_name, model_outputs); in runVirtualTempEstimator()
1123 *outputs = model_outputs; in runVirtualTempEstimator()
1142 model_outputs.clear(); in runVirtualTempEstimator()
1143 model_outputs.push_back(backup_sensor_vt); in runVirtualTempEstimator()
1144 *outputs = model_outputs; in runVirtualTempEstimator()
/aosp_15_r20/external/pytorch/aten/src/ATen/nnapi/
H A Dnnapi_model_loader.cpp242 const uint32_t* model_outputs = (const uint32_t*)next_pointer; in load_nnapi_model() local
251 model_outputs); in load_nnapi_model()
/aosp_15_r20/external/tensorflow/tensorflow/lite/testing/nnapi_tflite_zip_tests/
H A Dparse_testdata.cc195 int model_outputs = interpreter->outputs().size(); in CheckOutputs() local
196 TF_LITE_ENSURE_EQ(context, model_outputs, example.outputs.size()); in CheckOutputs()
/aosp_15_r20/external/executorch/backends/apple/coreml/runtime/delegate/
H A DETCoreMLModelManager.mm112 void set_outputs(NSArray<MLMultiArray *> *outputs, NSArray<MLMultiArray *> *model_outputs) {
113 NSEnumerator<MLMultiArray *> *enumerator = [model_outputs objectEnumerator];
152 NSArray<MLMultiArray *> *model_outputs) {
153 NSEnumerator<MLMultiArray *> *enumerator = [model_outputs objectEnumerator];
/aosp_15_r20/external/tensorflow/tensorflow/lite/testing/
H A Dparse_testdata.cc197 int model_outputs = interpreter->outputs().size(); in CheckOutputs() local
198 TF_LITE_ENSURE_EQ(context, model_outputs, example.outputs.size()); in CheckOutputs()
/aosp_15_r20/external/executorch/backends/apple/coreml/runtime/sdk/
H A DETCoreMLModelProfiler.mm207 NSArray<MLMultiArray *> *_Nullable __autoreleasing *_Nonnull model_outputs) {
217 *model_outputs = values;
H A DETCoreMLModelDebugger.mm349 NSArray<MLMultiArray *> *_Nullable __autoreleasing *_Nonnull model_outputs) {
359 *model_outputs = values;
/aosp_15_r20/packages/modules/NeuralNetworks/runtime/test/
DTestValidateModel.cpp3054 const uint32_t model_outputs[] = {309, 244, 245, 305}; in TEST_F() local
3056 ANeuralNetworksModel_identifyInputsAndOutputs(model, 4, model_inputs, 4, model_outputs), in TEST_F()