Home
last modified time | relevance | path

Searched defs:FillFunctor (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DFillKernel.cu13 struct FillFunctor { struct
14 FillFunctor(scalar_t v): value(v) {} in FillFunctor() argument
15 __device__ __forceinline__ scalar_t operator() () const { in operator ()()
19 scalar_t value;
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dfill_functor.cc107 struct FillFunctor<Eigen::ThreadPoolDevice, T> { struct
108 void operator()(const Eigen::ThreadPoolDevice& d, in operator ()()
H A Dfill_functor.cu.cc71 struct FillFunctor<GPUDevice, T> { struct
72 void operator()(const GPUDevice& d, typename TTypes<T>::Flat out, in operator ()()
H A Dfill_functor.h29 struct FillFunctor { struct