Skip to content
Snippets Groups Projects
Commit 259b6d6e authored by laurent soulier's avatar laurent soulier
Browse files

[MERGE][origin/master][NF] broken merge, issue with Slice

parent 812c08bc
No related branches found
No related tags found
1 merge request!9Refactoring Tensor
...@@ -29,7 +29,7 @@ void SliceImpl_cpu_forward_kernel(const typename Slice_Op::Attrs& attrs, ...@@ -29,7 +29,7 @@ void SliceImpl_cpu_forward_kernel(const typename Slice_Op::Attrs& attrs,
const I* input = static_cast<const I*>(input_) + std::get<0>(attrs); const I* input = static_cast<const I*>(input_) + std::get<0>(attrs);
I* output = static_cast<I*>(output_); I* output = static_cast<I*>(output_);
const std::vector<DimSize_t> slicedDims = std::get<1>(attrs); const std::vector<Coord_t> slicedDims = std::get<1>(attrs);
const std::size_t nbDims = slicedDims.size(); const std::size_t nbDims = slicedDims.size();
// for inputDims = {4,5,5,3} & slicedDims = {3,2,2,1}, substractDims = {1,5,5,3} // for inputDims = {4,5,5,3} & slicedDims = {3,2,2,1}, substractDims = {1,5,5,3}
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment