summaryrefslogtreecommitdiffstats
path: root/eval/src/tests/instruction/generic_peek/generic_peek_test.cpp
diff options
context:
space:
mode:
authorArne Juul <arnej@verizonmedia.com>2020-11-30 13:15:27 +0000
committerArne Juul <arnej@verizonmedia.com>2020-11-30 13:16:19 +0000
commitc65ddeb8386bda44c33324a3f6e8950a817ad46d (patch)
tree65cc76189385c3bf793141eb66406283728af37e /eval/src/tests/instruction/generic_peek/generic_peek_test.cpp
parent78925d049da67cd7d6adedef7fcc5b908fe21e58 (diff)
renumber Peek children, the sequel
* change the ReferenceOperations peek API also, putting the input parameter as the first child, and numbering child indexes in the Spec starting at 1.
Diffstat (limited to 'eval/src/tests/instruction/generic_peek/generic_peek_test.cpp')
-rw-r--r--eval/src/tests/instruction/generic_peek/generic_peek_test.cpp3
1 files changed, 2 insertions, 1 deletions
diff --git a/eval/src/tests/instruction/generic_peek/generic_peek_test.cpp b/eval/src/tests/instruction/generic_peek/generic_peek_test.cpp
index 08e445ca701..6841215038a 100644
--- a/eval/src/tests/instruction/generic_peek/generic_peek_test.cpp
+++ b/eval/src/tests/instruction/generic_peek/generic_peek_test.cpp
@@ -39,6 +39,7 @@ using PeekSpec = GenericPeek::SpecMap;
TensorSpec reference_peek(const TensorSpec &param, const PeekSpec &spec) {
std::vector<TensorSpec> children;
+ children.push_back(param);
PeekSpec with_indexes;
for (const auto & [dim_name, label_or_child] : spec) {
const vespalib::string &dim = dim_name;
@@ -59,7 +60,7 @@ TensorSpec reference_peek(const TensorSpec &param, const PeekSpec &spec) {
}
}, label_or_child);
}
- return ReferenceOperations::peek(param, with_indexes, children);
+ return ReferenceOperations::peek(with_indexes, children);
}
TensorSpec perform_generic_peek(const TensorSpec &a, const ValueType &result_type,