diff options
author | Arne Juul <arnej@verizonmedia.com> | 2020-11-30 13:15:27 +0000 |
---|---|---|
committer | Arne Juul <arnej@verizonmedia.com> | 2020-11-30 13:16:19 +0000 |
commit | c65ddeb8386bda44c33324a3f6e8950a817ad46d (patch) | |
tree | 65cc76189385c3bf793141eb66406283728af37e /eval/src/tests/instruction/generic_peek/generic_peek_test.cpp | |
parent | 78925d049da67cd7d6adedef7fcc5b908fe21e58 (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.cpp | 3 |
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 ¶m, 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 ¶m, 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, |