Searched refs:nc_ (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_()
44 EIGEN_DEVICE_FUNC EIGEN_ALWAYS_INLINE Index nc() const { return nc_; } in nc()
49 Index nc_; variable
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | eigen_contraction_kernel.h | 193 : kc_(k), mc_(m), nc_(n) {
196 computeProductBlockingSizes<Scalar, Scalar, 1>(kc_, mc_, nc_,
199 computeProductBlockingSizes<Scalar, Scalar, 1>(kc_, nc_, mc_,
204 if (kc_ <= 0 || mc_ <= 0 || nc_ <= 0) return;
214 nc_ = (std::min)(
215 n, Eigen::divup(static_cast<StorageIndex>(nc_ * kScaleN), kUnrollN) *
230 EIGEN_ALWAYS_INLINE StorageIndex nc() const { return nc_; }
235 StorageIndex nc_;
|