Home
last modified time | relevance | path

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

/external/eigen/unsupported/Eigen/CXX11/src/Tensor/
DTensorChipping.h89 : m_xpr(expr), m_offset(offset), m_dim(dim) {
95 const Index dim() const { return m_dim.actualDim(); }
123 const internal::DimensionId<DimId> m_dim;
153 : m_impl(op.expression(), device), m_dim(op.dim()), m_device(device)
156 eigen_assert(NumInputDims > m_dim.actualDim());
159 eigen_assert(op.offset() < input_dims[m_dim.actualDim()]);
163 if (i != m_dim.actualDim()) {
172 for (int i = 0; i < m_dim.actualDim(); ++i) {
177 for (int i = NumInputDims-1; i > m_dim.actualDim(); --i) {
182 m_inputStride *= input_dims[m_dim.actualDim()];
[all …]
/external/tensorflow/tensorflow/core/grappler/costs/
Dop_level_cost_estimator.cc704 double m_dim, n_dim, k_dim, k_dim_b = 0; in CountMatMulOperations() local
721 m_dim = a_matrix_shape.dim(1).size(); in CountMatMulOperations()
724 m_dim = a_matrix_shape.dim(0).size(); in CountMatMulOperations()
735 VLOG(1) << "M, N, K: " << m_dim << "," << n_dim << "," << k_dim; in CountMatMulOperations()
746 ops = m_dim * n_dim * k_dim * 2; in CountMatMulOperations()
750 mat_mul->m = m_dim; in CountMatMulOperations()
/external/tensorflow/tensorflow/compiler/xla/service/
Dtriangular_solve_expander.cc266 int64 m_dim = (left_side) ? -1 : -2; in SolveWithInvertedDiagonalBlocks() local
267 int64 m = ShapeUtil::GetDimension(b_shape, m_dim); in SolveWithInvertedDiagonalBlocks()