Searched refs:mc_ (Results 1 – 2 of 2) sorted by relevance
| /external/eigen/unsupported/Eigen/CXX11/src/Tensor/ |
| D | TensorContractionBlocking.h | 32 kc_(k), mc_(m), nc_(n) in kc_()
35 computeProductBlockingSizes<LhsScalar, RhsScalar, 1>(kc_, mc_, nc_, num_threads); in kc_()
38 computeProductBlockingSizes<LhsScalar, RhsScalar, 1>(kc_, nc_, mc_, num_threads); in kc_()
43 EIGEN_DEVICE_FUNC EIGEN_ALWAYS_INLINE Index mc() const { return mc_; } in mc()
48 Index mc_; variable
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | eigen_contraction_kernel.h | 296 : kc_(k), mc_(m), nc_(n) {
299 computeProductBlockingSizes<Scalar, Scalar, 1>(kc_, mc_, nc_,
302 computeProductBlockingSizes<Scalar, Scalar, 1>(kc_, nc_, mc_,
307 if (kc_ <= 0 || mc_ <= 0 || nc_ <= 0) return;
314 mc_ = (std::min)(
315 m, Eigen::divup(static_cast<StorageIndex>(mc_ * kScaleM), kUnrollM) *
332 EIGEN_ALWAYS_INLINE StorageIndex mc() const { return mc_; }
337 StorageIndex mc_;
362 : kc_(k), mc_(m), nc_(n) {
364 mc_ = (std::min)(m, static_cast<StorageIndex>(192));
[all …]
|