Searched refs:SetZeroFunctor (Results 1 – 20 of 20) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | fill_functor.h | 36 struct SetZeroFunctor { struct
43 struct SetZeroFunctor<Eigen::ThreadPoolDevice, T> {
51 struct SetZeroFunctor<Eigen::SyclDevice, T> {
57 struct SetZeroFunctor<Eigen::ThreadPoolDevice, string> {
|
| D | fill_functor.cc | 30 void SetZeroFunctor<Eigen::ThreadPoolDevice, T>::operator()( in operator ()()
35 void SetZeroFunctor<Eigen::ThreadPoolDevice, string>::operator()( in operator ()()
42 template struct SetZeroFunctor<Eigen::ThreadPoolDevice, T>;
66 void SetZeroFunctor<Eigen::SyclDevice, T>::operator()( in operator ()()
72 template struct SetZeroFunctor<Eigen::SyclDevice, T>;
|
| D | fill_functor.cu.cc | 85 struct SetZeroFunctor<GPUDevice, T> { struct
92 void SetZeroFunctor<GPUDevice, Variant>::operator()( in operator ()()
97 #define DEFINE_SETZERO_GPU(T) template struct SetZeroFunctor<GPUDevice, T>;
|
| D | constant_op_gpu.cu.cc | 85 struct SetZeroFunctor<GPUDevice, T> { struct
91 #define DEFINE_SETZERO_GPU(T) template struct SetZeroFunctor<GPUDevice, T>;
|
| D | conditional_accumulator.h | 67 functor::SetZeroFunctor<Device, T> set_zero_functor_;
|
| D | tensor_array.cc | 59 functor::SetZeroFunctor<Device, T> set_zero_functor; \
|
| D | list_kernels.h | 158 functor::SetZeroFunctor<Device, T>()(c->eigen_device<Device>(), in Compute()
238 functor::SetZeroFunctor<Device, T>()(c->eigen_device<Device>(), in Compute()
285 functor::SetZeroFunctor<Device, T>()(c->eigen_device<Device>(), in Compute()
472 functor::SetZeroFunctor<Device, T>()(c->eigen_device<Device>(), in Compute()
647 functor::SetZeroFunctor<Device, T>()(c->eigen_device<Device>(), in Compute()
|
| D | mkl_matmul_op.cc | 87 functor::SetZeroFunctor<Device, T> f; in Compute()
|
| D | mkl_batch_matmul_op.cc | 97 functor::SetZeroFunctor<Device, Scalar> f; in Compute()
|
| D | sparse_tensor_dense_matmul_op.cc | 130 functor::SetZeroFunctor<Device, T> f; in Compute()
|
| D | fused_batch_norm_op.cc | 662 functor::SetZeroFunctor<Device, float> f; in Compute()
671 functor::SetZeroFunctor<Device, U> f; in Compute()
|
| D | determinant_op.cc | 306 functor::SetZeroFunctor<GPUDevice, Scalar> zero_func; in ComputeAsync()
|
| D | constant_op.cc | 275 functor::SetZeroFunctor<Device, T> f; in Compute()
|
| D | dynamic_partition_op_gpu.cu.cc | 379 functor::SetZeroFunctor<GPUDevice, int32> zero_functor; in CountAndSortParts()
|
| D | inplace_ops.cc | 452 functor::SetZeroFunctor<Device, T>()(ctx->eigen_device<Device>(), in Compute()
|
| D | matmul_op.cc | 490 functor::SetZeroFunctor<Device, T> f; in Compute()
|
| D | batch_matmul_op_impl.h | 586 functor::SetZeroFunctor<Device, Scalar> f;
|
| D | scatter_nd_op.cc | 687 functor::SetZeroFunctor<Device, T> fill; in DoScatterNd()
|
| D | conv_grad_filter_ops.cc | 512 functor::SetZeroFunctor<Device, T> f; in Compute()
|
| D | sparse_matmul_op.cc | 990 functor::SetZeroFunctor<CPUDevice, float> f; in Compute()
|