Searched refs:AsEigenDSizes (Results 1 – 6 of 6) sorted by relevance
83 functor(context->eigen_device<Device>(), shape_in.AsEigenDSizes<2>(), in Compute()89 functor(context->eigen_device<Device>(), shape_in.AsEigenDSizes<2>(), in Compute()
154 in.shape().AsEigenDSizes<NDIMS>()); in TransposeUsingEigen()157 out->shape().AsEigenDSizes<NDIMS>()); in TransposeUsingEigen()
652 indices_host_, indices.shape().AsEigenDSizes<2>()); in operator ()()
719 shape().AsEigenDSizes<NDIMS>()); in tensor()726 shape().AsEigenDSizes<NDIMS>()); in tensor()733 shape().AsEigenDSizes<NDIMS>()); in bit_casted_tensor()740 shape().AsEigenDSizes<NDIMS>()); in bit_casted_tensor()
307 Eigen::DSizes<IndexType, NDIMS> AsEigenDSizes() const;467 Eigen::DSizes<IndexType, NDIMS> TensorShape::AsEigenDSizes() const { in AsEigenDSizes() function
294 Eigen::DSizes<Eigen::DenseIndex, NDIMS> AsEigenDSizes() const;