Searched refs:FillFunctor (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | fill_functor.cc | 125 struct FillFunctor<Eigen::ThreadPoolDevice, T> { struct
135 template struct FillFunctor<Eigen::ThreadPoolDevice, T>;
144 struct FillFunctor<Eigen::SyclDevice, T> { struct
159 #define DEFINE_FILL_SYCL(T) template struct FillFunctor<Eigen::SyclDevice, T>;
|
| D | fill_functor.cu.cc | 70 struct FillFunctor<GPUDevice, T> { struct
78 #define DEFINE_FILL_GPU(T) template struct FillFunctor<GPUDevice, T>;
|
| D | constant_op_gpu.cu.cc | 70 struct FillFunctor<GPUDevice, T> { struct
78 #define DEFINE_FILL_GPU(T) template struct FillFunctor<GPUDevice, T>;
|
| D | fill_functor.h | 29 struct FillFunctor { struct
|
| D | broadcast_to_op.cc | 72 functor::FillFunctor<Device, T>()(device, output_tensor->flat<T>(), in Compute()
|
| D | constant_op.cc | 171 functor::FillFunctor<Device, T> functor; in Compute()
|