Searched refs:FillFunctor (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | constant_op_gpu.cu.cc | 71 struct FillFunctor<GPUDevice, T> { struct
79 #define DEFINE_FILL_GPU(T) template struct FillFunctor<GPUDevice, T>;
|
| D | fill_functor.cc | 95 struct FillFunctor<Eigen::ThreadPoolDevice, T> { struct
105 template struct FillFunctor<Eigen::ThreadPoolDevice, T>;
|
| D | fill_functor.cu.cc | 71 struct FillFunctor<GPUDevice, T> { struct
79 #define DEFINE_FILL_GPU(T) template struct FillFunctor<GPUDevice, T>;
|
| D | fill_functor.h | 29 struct FillFunctor { struct
|
| D | broadcast_to_op.cc | 74 functor::FillFunctor<Device, T>()(device, output_tensor->flat<T>(), in Compute()
|
| D | constant_op.cc | 162 functor::FillFunctor<Device, T> functor; in Compute()
|