Searched refs:m_dims (Results 1 – 6 of 6) sorted by relevance
| /external/eigen/unsupported/Eigen/CXX11/src/Tensor/ |
| D | TensorEvaluator.h | 49 …ternal::traits<Derived>::template MakePointer<Scalar>::Type>(m.data())), m_dims(m.dimensions()), m… in TensorEvaluator()
54 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const Dimensions& dimensions() const { return m_dims; } in dimensions()
58 m_device.memcpy((void*)dest, m_data, sizeof(Scalar) * m_dims.TotalSize()); in evalSubExprsIfNeeded()
91 return m_data[m_dims.IndexOfColMajor(coords)]; in coeff()
93 return m_data[m_dims.IndexOfRowMajor(coords)]; in coeff()
100 return m_data[m_dims.IndexOfColMajor(coords)]; in coeffRef()
102 return m_data[m_dims.IndexOfRowMajor(coords)]; in coeffRef()
118 Dimensions m_dims; member
172 : m_data(m.data()), m_dims(m.dimensions()), m_device(device), m_impl(m)
175 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const Dimensions& dimensions() const { return m_dims; }
[all …]
|
| D | TensorStriding.h | 64 : m_xpr(expr), m_dims(dims) {}
67 const Strides& strides() const { return m_dims; }
94 const Strides m_dims;
|
| D | TensorRef.h | 49 m_dims = m_impl.dimensions(); in TensorLazyEvaluatorReadOnly()
57 return m_dims; in dimensions()
73 Dimensions m_dims; variable
|
| D | TensorReduction.h | 360 TensorReductionOp(const XprType& expr, const Dims& dims) : m_expr(expr), m_dims(dims)
363 …onst XprType& expr, const Dims& dims, const Op& reducer) : m_expr(expr), m_dims(dims), m_reducer(r…
369 const Dims& dims() const { return m_dims; }
375 const Dims m_dims;
|
| D | TensorMorphing.h | 63 : m_xpr(expr), m_dims(dims) {}
66 const NewDimensions& dimensions() const { return m_dims; }
93 const NewDimensions m_dims;
|
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_fused_batch_norm_op.cc | 212 memory::dims m_dims = {1, fwdParams.depth}; in Setup() local
221 new memory({{m_dims}, MklDnnType<U>(), memory::format_tag::nc}, in Setup()
225 new memory({{m_dims}, MklDnnType<U>(), memory::format_tag::nc}, in Setup()
|