| /external/tensorflow/tensorflow/core/kernels/ |
| D | fill_functor.h | 36 struct SetZeroFunctor { struct
43 struct SetZeroFunctor<Eigen::ThreadPoolDevice, T> {
50 struct SetZeroFunctor<Eigen::ThreadPoolDevice, tstring> {
|
| D | fill_functor.cc | 30 void SetZeroFunctor<Eigen::ThreadPoolDevice, T>::operator()( in operator ()()
35 void SetZeroFunctor<Eigen::ThreadPoolDevice, tstring>::operator()( in operator ()()
42 template struct SetZeroFunctor<Eigen::ThreadPoolDevice, T>;
|
| D | fill_functor.cu.cc | 86 struct SetZeroFunctor<GPUDevice, T> { struct
93 void SetZeroFunctor<GPUDevice, Variant>::operator()( in operator ()()
98 #define DEFINE_SETZERO_GPU(T) template struct SetZeroFunctor<GPUDevice, T>;
|
| D | constant_op_gpu.cu.cc | 86 struct SetZeroFunctor<GPUDevice, T> { struct
92 #define DEFINE_SETZERO_GPU(T) template struct SetZeroFunctor<GPUDevice, T>;
|
| D | bincount_op.cc | 244 functor::SetZeroFunctor<Device, T> fill; in Compute()
342 functor::SetZeroFunctor<Device, T> fill; in Compute()
426 functor::SetZeroFunctor<Device, T> fill; in Compute()
|
| D | tensor_array.cc | 59 functor::SetZeroFunctor<Device, T> set_zero_functor; \
|
| D | conditional_accumulator.h | 67 functor::SetZeroFunctor<Device, T> set_zero_functor_;
|
| D | list_kernels.h | 137 functor::SetZeroFunctor<Device, T>()(c->eigen_device<Device>(), in Compute()
217 functor::SetZeroFunctor<Device, T>()(c->eigen_device<Device>(), in Compute()
264 functor::SetZeroFunctor<Device, T>()(c->eigen_device<Device>(), in Compute()
449 functor::SetZeroFunctor<Device, T>()(c->eigen_device<Device>(), in Compute()
624 functor::SetZeroFunctor<Device, T>()(c->eigen_device<Device>(), in Compute()
|
| D | matmul_op_fused.cc | 212 functor::SetZeroFunctor<Device, T> f; in Compute()
|
| D | sparse_tensor_dense_matmul_op.cc | 131 functor::SetZeroFunctor<Device, T> f; in Compute()
|
| D | constant_op.cc | 251 functor::SetZeroFunctor<Device, T> f; in Compute()
|
| D | stateful_random_ops.cc | 292 functor::SetZeroFunctor<Device, T>()(ctx->eigen_device<Device>(), in Compute()
|
| D | inplace_ops.cc | 401 functor::SetZeroFunctor<Device, T>()(ctx->eigen_device<Device>(), in Compute()
|
| D | dynamic_partition_op_gpu.cu.cc | 389 functor::SetZeroFunctor<GPUDevice, int32> zero_functor; in CountAndSortParts()
|
| /external/tensorflow/tensorflow/core/kernels/sparse/ |
| D | zeros_op.h | 68 functor::SetZeroFunctor<Device, int32> set_zero; in operator()
|
| D | sparse_tensor_to_csr_sparse_matrix_op.cc | 85 functor::SetZeroFunctor<CPUDevice, int32> set_zero; in Compute()
242 functor::SetZeroFunctor<Device, int32> set_zero; in ComputeAsync()
|
| D | dense_to_csr_sparse_matrix_op.cc | 109 functor::SetZeroFunctor<Device, int32> set_zero; in Compute()
309 functor::SetZeroFunctor<Device, int32> set_zero; in ComputeAsync()
|
| D | csr_sparse_matrix_to_dense_op.cc | 89 functor::SetZeroFunctor<Device, T> set_zero; in Compute()
|
| D | transpose_op.cc | 188 functor::SetZeroFunctor<Device, int32> set_zero; in operator ()()
|
| D | add_op.cc | 106 functor::SetZeroFunctor<Device, int32> set_zero; in operator ()()
|
| D | mat_mul_op.cc | 362 functor::SetZeroFunctor<CPUDevice, T> set_zero; in SparseDenseMatMulWithTransposedLHS()
|
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_matmul_op.cc | 84 functor::SetZeroFunctor<Device, T> f; in Compute()
|
| D | mkl_batch_matmul_op.cc | 134 functor::SetZeroFunctor<Device, Scalar> f; in Compute()
|
| /external/tensorflow/tensorflow/core/kernels/linalg/ |
| D | determinant_op.cc | 306 functor::SetZeroFunctor<GPUDevice, Scalar> zero_func; in ComputeAsync()
|
| D | einsum_op_impl.h | 575 functor::SetZeroFunctor<Device, T> set_zero; in ContractOperands()
|