Block evaluation for TensorGenerator + TensorReverse + fixed bug in tensor reverse op

This commit is contained in:
Eugene Zhulenev
2019-10-10 10:56:58 -07:00
parent b03eb63d7c
commit a411e9f344
5 changed files with 303 additions and 40 deletions

View File

@@ -131,7 +131,7 @@ struct TensorEvaluator<const TensorEvalToOp<ArgType, MakePointer_>, Device>
ArgTensorBlock;
typedef internal::TensorBlockAssignment<
Scalar, NumDims, typename ArgTensorBlock::XprType, Index>
CoeffReturnType, NumDims, typename ArgTensorBlock::XprType, Index>
TensorBlockAssignment;
//===--------------------------------------------------------------------===//