devtools/eigen
1
0
Fork 0
mirror of https://gitlab.com/libeigen/eigen.git synced 2026-04-10 11:34:33 +08:00
Code Issues Packages Projects Releases Wiki Activity

Add block evaluation to TensorReshaping/TensorCasting/TensorPadding/TensorSelect

Browse Source
This commit is contained in:
Eugene Zhulenev
2019-10-02 12:44:06 -07:00
parent 6e40454a6e
commit 60ae24ee1a
9 changed files with 862 additions and 167 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
unsupported/Eigen/CXX11/src/Tensor/TensorBroadcasting.h
View File

@@ -1247,10 +1247,10 @@ struct TensorEvaluator<const TensorBroadcastingOp<Broadcast, ArgType>, Device>
ScalarNoConst, NumDims, typename ArgTensorBlock::XprType, Index>
TensorBlockAssignment;
typename TensorBlockAssignment::Dst assignment_dst(
input_block_sizes, input_block_strides, *materialized_input);
TensorBlockAssignment::Run(assignment_dst, input_block.expr());
TensorBlockAssignment::Run(
TensorBlockAssignment::target(input_block_sizes, input_block_strides,
*materialized_input),
input_block.expr());
input_buffer = *materialized_input;
}