| /external/tensorflow/tensorflow/core/kernels/ |
| fill_functor.h | 36 struct SetZeroFunctor {
41 // Partial specialization of SetZeroFunctor<Device=Eigen::ThreadPoolDevice, T>.
43 struct SetZeroFunctor<Eigen::ThreadPoolDevice, T> {
49 // Partial specialization of SetZeroFunctor<Device=Eigen::SyclDevice, T>.
51 struct SetZeroFunctor<Eigen::SyclDevice, T> {
57 struct SetZeroFunctor<Eigen::ThreadPoolDevice, string> {
|
| fill_functor.cc | 30 void SetZeroFunctor<Eigen::ThreadPoolDevice, T>::operator()(
35 void SetZeroFunctor<Eigen::ThreadPoolDevice, string>::operator()(
42 template struct SetZeroFunctor<Eigen::ThreadPoolDevice, T>;
61 void SetZeroFunctor<Eigen::SyclDevice, T>::operator()(
67 template struct SetZeroFunctor<Eigen::SyclDevice, T>;
|
| constant_op_gpu.cu.cc | 85 struct SetZeroFunctor<GPUDevice, T> {
91 #define DEFINE_SETZERO_GPU(T) template struct SetZeroFunctor<GPUDevice, T>;
|
| fill_functor.cu.cc | 85 struct SetZeroFunctor<GPUDevice, T> {
91 #define DEFINE_SETZERO_GPU(T) template struct SetZeroFunctor<GPUDevice, T>;
|
| conditional_accumulator.h | 65 functor::SetZeroFunctor<Device, T> set_zero_functor_;
|
| tensor_array.cc | 59 functor::SetZeroFunctor<Device, T> set_zero_functor; \
|
| mkl_matmul_op.cc | 81 functor::SetZeroFunctor<Device, T> f;
|
| mkl_batch_matmul_op.cc | 97 functor::SetZeroFunctor<Device, Scalar> f;
|
| constant_op.cc | 296 functor::SetZeroFunctor<Device, T> f;
|
| determinant_op.cc | 306 functor::SetZeroFunctor<GPUDevice, Scalar> zero_func;
|
| sparse_tensor_dense_matmul_op.cc | 130 functor::SetZeroFunctor<Device, T> f;
|
| fused_batch_norm_op.cc | 641 functor::SetZeroFunctor<Device, float> f;
650 functor::SetZeroFunctor<Device, U> f;
[all...] |
| batch_matmul_op_impl.h | 494 functor::SetZeroFunctor<Device, Scalar> f;
|
| dynamic_partition_op_gpu.cu.cc | 378 functor::SetZeroFunctor<GPUDevice, int32> zero_functor;
|
| matmul_op.cc | 491 functor::SetZeroFunctor<Device, T> f;
|
| scatter_nd_op.cc | 462 functor::SetZeroFunctor<Device, T> fill;
|
| conv_grad_filter_ops.cc | 601 functor::SetZeroFunctor<Device, T> f;
[all...] |
| sparse_matmul_op.cc | [all...] |
