MiniTensor Version of the Day
Loading...
Searching...
No Matches
MiniTensor_TensorBase.h File Reference
#include <algorithm>
#include <cassert>
#include <iostream>
#include <vector>
#include "MiniTensor_Storage.h"
#include "MiniTensor_Scalar.h"

Go to the source code of this file.

Classes

class  minitensor::TensorBase< T, ST >
 

Namespaces

namespace  minitensor
 

Enumerations

enum class  minitensor::Filler {
  minitensor::Filler::ZEROS , minitensor::Filler::ONES , minitensor::Filler::SEQUENCE , minitensor::Filler::RANDOM ,
  minitensor::Filler::RANDOM_UNIFORM , minitensor::Filler::RANDOM_NORMAL , minitensor::Filler::NANS
}
 
enum class  minitensor::ComponentOrder { minitensor::ComponentOrder::CANONICAL , minitensor::ComponentOrder::SIERRA_FULL , minitensor::ComponentOrder::SIERRA_SYMMETRIC }
 
enum class  minitensor::Source { minitensor::Source::ARRAY }
 

Functions

template<typename T , typename ST >
KOKKOS_INLINE_FUNCTIONminitensor::norm_f (TensorBase< T, ST > const &X)
 
template<typename T , typename ST >
KOKKOS_INLINE_FUNCTIONminitensor::norm_f_square (TensorBase< T, ST > const &X)
 
template<typename R , typename S , typename T , typename SR , typename SS , typename ST >
KOKKOS_INLINE_FUNCTION void minitensor::add (TensorBase< R, SR > const &A, TensorBase< S, SS > const &B, TensorBase< T, ST > &C)
 
template<typename R , typename S , typename T , typename SR , typename SS , typename ST >
KOKKOS_INLINE_FUNCTION void minitensor::subtract (TensorBase< R, SR > const &A, TensorBase< S, SS > const &B, TensorBase< T, ST > &C)
 
template<typename T , typename ST >
KOKKOS_INLINE_FUNCTION void minitensor::minus (TensorBase< T, ST > const &A, TensorBase< T, ST > &B)
 
template<typename T , typename ST >
KOKKOS_INLINE_FUNCTION bool minitensor::equal (TensorBase< T, ST > const &A, TensorBase< T, ST > const &B)
 
template<typename T , typename ST >
KOKKOS_INLINE_FUNCTION bool minitensor::not_equal (TensorBase< T, ST > const &A, TensorBase< T, ST > const &B)
 
template<typename R , typename S , typename T , typename SR , typename ST >
KOKKOS_INLINE_FUNCTION void minitensor::scale (TensorBase< R, SR > const &A, S const &s, TensorBase< T, ST > &B)
 
template<typename R , typename S , typename T , typename SR , typename ST >
KOKKOS_INLINE_FUNCTION void minitensor::divide (TensorBase< R, SR > const &A, S const &s, TensorBase< T, ST > &B)
 
template<typename R , typename S , typename T , typename SR , typename ST >
KOKKOS_INLINE_FUNCTION void minitensor::split (TensorBase< R, SR > const &A, S const &s, TensorBase< T, ST > &B)