Home
last modified time | relevance | path

Searched refs:TensorBlockKind (Results 1 – 7 of 7) sorted by relevance

/external/eigen/unsupported/Eigen/CXX11/src/Tensor/
DTensorBlock.h589 enum TensorBlockKind { enum
661 TensorMaterializedBlock(TensorBlockKind kind, const Scalar* data,
668 eigen_assert(m_kind == internal::TensorBlockKind::kView ||
669 m_kind == internal::TensorBlockKind::kMaterializedInScratch ||
670 m_kind == internal::TensorBlockKind::kMaterializedInOutput);
673 TensorBlockKind kind() const { return m_kind; }
703 ? internal::TensorBlockKind::kMaterializedInOutput
704 : internal::TensorBlockKind::kMaterializedInScratch,
800 return TensorMaterializedBlock(internal::TensorBlockKind::kView,
823 TensorBlockKind m_kind;
[all …]
DTensorEvalTo.h184 if (block.kind() != internal::TensorBlockKind::kMaterializedInOutput) {
DTensorAssign.h223 if (block.kind() != internal::TensorBlockKind::kMaterializedInOutput) {
DTensorPadding.h243 return TensorBlock(internal::TensorBlockKind::kView, NULL,
DTensorBroadcasting.h846 return TensorBlock(internal::TensorBlockKind::kView, NULL, dimensions);
DTensorMorphing.h210 return TensorBlock(internal::TensorBlockKind::kView,
/external/eigen/unsupported/test/
Dcxx11_tensor_block_eval.cpp164 if (tensor_block.kind() == internal::TensorBlockKind::kMaterializedInOutput) { in VerifyBlockEvaluator()
613 internal::TensorBlockKind::kView, block.data(), block.dimensions()); in VerifyBlockAssignment()