Home
last modified time | relevance | path

Searched refs:is_rhs (Results 1 – 3 of 3) sorted by relevance

/external/eigen/unsupported/Eigen/CXX11/src/Tensor/
DTensorContractionSycl.h321 static EIGEN_CONSTEXPR bool is_rhs = is_rhs_; member
324 static EIGEN_CONSTEXPR bool is_coalesced_layout = !(is_transposed ^ is_rhs);
717 InputBlockProperties::is_rhs ? Properties::LocalThreadSizeN : Properties::LocalThreadSizeM;
719 … InputBlockProperties::is_rhs ? Properties::WorkLoadPerThreadN : Properties::WorkLoadPerThreadM;
720 const StorageIndex &NC = InputBlockProperties::is_rhs ? triple_dim.N : triple_dim.M;
737 … InputBlockProperties::is_rhs, typename InputBlockProperties::OutType>(inpt, ncIndex, cIndex, ld);
748 … ? read<false, InputBlockProperties::is_coalesced_layout, InputBlockProperties::is_rhs, OutScalar>(
760 …ncIndex = (!InputBlockProperties::is_rhs && InputBlockProperties::nc_stride == 1 && PacketSize != …
901 InputBlockProperties::is_rhs ? Properties::TileSizeDimN : Properties::TileSizeDimM;
903 InputBlockProperties::is_rhs ? Properties::LoadPerThreadRhs : Properties::LoadPerThreadLhs;
[all …]
DTensorContractionThreadPool.h695 template <typename BlockType, bool is_rhs>
698 !is_rhs && std::is_same<BlockType, LhsBlock>::value;
700 is_rhs && std::is_same<BlockType, RhsBlock>::value;
715 ThreadLocalBlocksAllocator<is_rhs>::allocate(ctx_, blocks);
717 ThreadLocalBlocksAllocator<is_rhs>::reuse(ctx_, n, blocks);
/external/tensorflow/tensorflow/compiler/xla/service/
Dspace_to_batch_converter.cc132 bool is_backprop = false, bool is_rhs = false);
207 bool is_backprop = false, bool is_rhs = false);
643 bool is_rhs) { in BringSpaceNextToBatch() argument
660 if (is_rhs) { in BringSpaceNextToBatch()
2940 bool is_rhs) { in SplitSpace() argument
2944 spatial_dimensions_to_split, is_backprop, is_rhs)); in SplitSpace()