HomeSort by relevance Sort by last modified time
    Searched defs:TensorMap (Results 1 - 3 of 3) sorted by null

  /external/eigen/unsupported/Eigen/CXX11/src/Tensor/
TensorMap.h 15 /** \class TensorMap
27 template<typename PlainObjectType, int Options_, template <class> class MakePointer_> class TensorMap : public TensorBase<TensorMap<PlainObjectType, Options_, MakePointer_> >
30 typedef TensorMap<PlainObjectType, Options_, MakePointer_> Self;
60 EIGEN_STRONG_INLINE TensorMap(PointerArgType dataPtr) : m_data(dataPtr), m_dimensions() {
67 EIGEN_STRONG_INLINE TensorMap(PointerArgType dataPtr, Index firstDimension, IndexTypes... otherDimensions) : m_data(dataPtr), m_dimensions(firstDimension, otherDimensions...) {
73 EIGEN_STRONG_INLINE TensorMap(PointerArgType dataPtr, Index firstDimension) : m_data(dataPtr), m_dimensions(firstDimension) {
78 EIGEN_STRONG_INLINE TensorMap(PointerArgType dataPtr, Index dim1, Index dim2) : m_data(dataPtr), m_dimensions(dim1, dim2) {
82 EIGEN_STRONG_INLINE TensorMap(PointerArgType dataPtr, Index dim1, Index dim2, Index dim3) : m_data(dataPtr), m_dimensions(dim1, dim2, dim3) {
86 EIGEN_STRONG_INLINE TensorMap(PointerArgType dataPtr, Index dim1, Index dim2, Index dim3, Index dim4) : m_data(dataPtr), m_dimen (…)
    [all...]
  /external/tensorflow/tensorflow/core/kernels/
conv_grad_filter_ops.cc 440 typedef Eigen::TensorMap<Eigen::Tensor<T, 2, Eigen::RowMajor>,
442 TensorMap;
443 typedef Eigen::TensorMap<Eigen::Tensor<const T, 2, Eigen::RowMajor>,
447 TensorMap C(filter_backprop_data, filter_total_size, dims.out_depth);
    [all...]
conv_grad_input_ops.cc 486 typedef Eigen::TensorMap<Eigen::Tensor<T, 2, Eigen::RowMajor>,
488 TensorMap;
489 typedef Eigen::TensorMap<Eigen::Tensor<const T, 2, Eigen::RowMajor>,
500 TensorMap C(col_buffer_data, output_image_size, filter_total_size);
    [all...]

Completed in 106 milliseconds