container::kernels::set_memory< T, Device > Struct Template Reference

A functor to set memory to a constant value. More...

#include <memory.h>

Collaboration diagram for container::kernels::set_memory< T, Device >:

Public Member Functions
void	operator() (T *arr, const T &var, const size_t &size)
	Set memory to a constant value.

Detailed Description

template<typename T, typename Device>
struct container::kernels::set_memory< T, Device >

A functor to set memory to a constant value.

Template Parameters

T	Floating-point type of the memory.
Device	Device type where the memory is allocated.

Member Function Documentation

operator()()

template<typename T , typename Device >

void container::kernels::set_memory< T, Device >::operator()	(	T *	arr,
		const T &	var,
		const size_t &	size
	)

Set memory to a constant value.

Parameters

arr	Pointer to the memory.
var	Constant value to set.
size	Size of the memory to set.

---

The documentation for this struct was generated from the following file:

• /home/runner/work/abacus-develop/abacus-develop/source/source_base/module_container/ATen/kernels/memory.h