From 90b0df73895a154ba1fec3cc8c67cbebc4823b3e Mon Sep 17 00:00:00 2001 From: HudsonGraeme Date: Sat, 4 Jul 2026 02:29:11 +0000 Subject: [PATCH 1/2] Introduce shaped output tensor accessor for named circuit outputs --- crates/dsperse/src/pipeline/combined.rs | 8 ++++++++ 1 file changed, 8 insertions(+) diff --git a/crates/dsperse/src/pipeline/combined.rs b/crates/dsperse/src/pipeline/combined.rs index 5aac2899..ba67d65a 100644 --- a/crates/dsperse/src/pipeline/combined.rs +++ b/crates/dsperse/src/pipeline/combined.rs @@ -261,6 +261,14 @@ impl CombinedRun { self.outputs_for_names(output_names) } + pub fn output_arrays_for_names(&self, names: &[String]) -> Option>> { + let mut arrays = Vec::with_capacity(names.len()); + for name in names { + arrays.push(self.tensor_cache.try_get(name)?.clone()); + } + if arrays.is_empty() { None } else { Some(arrays) } + } + pub fn outputs_for_names(&self, names: &[String]) -> Option> { let mut flat = Vec::new(); for name in names { From 3baa79ecbd29eafe9b2846176dc62a23c7ec3c07 Mon Sep 17 00:00:00 2001 From: HudsonGraeme Date: Sat, 4 Jul 2026 02:29:37 +0000 Subject: [PATCH 2/2] Conduct formatting normalization on output accessor --- crates/dsperse/src/pipeline/combined.rs | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/crates/dsperse/src/pipeline/combined.rs b/crates/dsperse/src/pipeline/combined.rs index ba67d65a..623d319a 100644 --- a/crates/dsperse/src/pipeline/combined.rs +++ b/crates/dsperse/src/pipeline/combined.rs @@ -266,7 +266,11 @@ impl CombinedRun { for name in names { arrays.push(self.tensor_cache.try_get(name)?.clone()); } - if arrays.is_empty() { None } else { Some(arrays) } + if arrays.is_empty() { + None + } else { + Some(arrays) + } } pub fn outputs_for_names(&self, names: &[String]) -> Option> {