/external/tensorflow/tensorflow/core/kernels/ |
reduction_ops_gpu_complex128.cu.cc | 25 typedef Eigen::GpuDevice GPUDevice; 37 template void ReduceFunctor<GPUDevice, REDUCER>::Reduce( \ 44 template void ReduceFunctor<GPUDevice, REDUCER>::FillIdentity( \ 45 const GPUDevice& d, TTypes<T>::Vec out, const REDUCER& reducer);
|
reduction_ops_gpu_complex64.cu.cc | 25 typedef Eigen::GpuDevice GPUDevice; 37 template void ReduceFunctor<GPUDevice, REDUCER>::Reduce( \ 44 template void ReduceFunctor<GPUDevice, REDUCER>::FillIdentity( \ 45 const GPUDevice& d, TTypes<T>::Vec out, const REDUCER& reducer);
|
reduction_ops_gpu_double.cu.cc | 25 typedef Eigen::GpuDevice GPUDevice; 37 template void ReduceFunctor<GPUDevice, REDUCER>::Reduce( \ 44 template void ReduceFunctor<GPUDevice, REDUCER>::FillIdentity( \ 45 const GPUDevice& d, TTypes<T>::Vec out, const REDUCER& reducer);
|
reduction_ops_gpu_float.cu.cc | 25 typedef Eigen::GpuDevice GPUDevice; 37 template void ReduceFunctor<GPUDevice, REDUCER>::Reduce( \ 44 template void ReduceFunctor<GPUDevice, REDUCER>::FillIdentity( \ 45 const GPUDevice& d, TTypes<T>::Vec out, const REDUCER& reducer);
|
reduction_ops_gpu_int.cu.cc | 25 typedef Eigen::GpuDevice GPUDevice; 37 template void ReduceFunctor<GPUDevice, REDUCER>::Reduce( \ 44 template void ReduceFunctor<GPUDevice, REDUCER>::FillIdentity( \ 45 const GPUDevice& d, TTypes<T>::Vec out, const REDUCER& reducer);
|
reduction_ops_half_mean_sum.cu.cc | 25 typedef Eigen::GpuDevice GPUDevice; 37 template void ReduceFunctor<GPUDevice, REDUCER>::Reduce( \ 44 template void ReduceFunctor<GPUDevice, REDUCER>::FillIdentity( \ 45 const GPUDevice& d, TTypes<T>::Vec out, const REDUCER& reducer);
|
reduction_ops_half_prod_max_min.cu.cc | 25 typedef Eigen::GpuDevice GPUDevice; 37 template void ReduceFunctor<GPUDevice, REDUCER>::Reduce( \ 44 template void ReduceFunctor<GPUDevice, REDUCER>::FillIdentity( \ 45 const GPUDevice& d, TTypes<T>::Vec out, const REDUCER& reducer);
|
scatter_functor.cc | 24 typedef Eigen::GpuDevice GPUDevice; 31 Index ScatterFunctor<GPUDevice, T, Index, op>::operator()( \ 32 OpKernelContext* c, const GPUDevice& d, \ 36 extern template struct ScatterFunctor<GPUDevice, T, Index, op>;
|
sparse_tensor_dense_matmul_op_gpu.cu.cc | 28 typedef Eigen::GpuDevice GPUDevice; 66 struct SparseTensorDenseMatMulFunctor<GPUDevice, T, Tindices, ADJ_A, ADJ_B> { 68 Compute(const GPUDevice& d, typename TTypes<T>::Matrix out, 97 GPUDevice, T, Tindices, false, false>; \ 99 GPUDevice, T, Tindices, false, true>; \ 101 GPUDevice, T, Tindices, true, false>; \ 103 GPUDevice, T, Tindices, true, true>;
|
diag_op.cc | 39 typedef Eigen::GpuDevice GPUDevice; 201 extern template struct DiagFunctor<GPUDevice, double>; 202 extern template struct DiagFunctor<GPUDevice, float>; 203 extern template struct DiagFunctor<GPUDevice, int32>; 204 extern template struct DiagFunctor<GPUDevice, int64>; 205 extern template struct DiagFunctor<GPUDevice, complex64>; 206 extern template struct DiagFunctor<GPUDevice, complex128>; 212 DiagOp<GPUDevice, T>) 224 extern template struct DiagPartFunctor<GPUDevice, double> [all...] |
colorspace_op.cc | 37 typedef Eigen::GpuDevice GPUDevice; 128 void RGBToHSV<GPUDevice, T>::operator()( \ 129 const GPUDevice& d, TTypes<T, 2>::ConstTensor input_data, \ 131 extern template struct RGBToHSV<GPUDevice, T>; \ 133 void HSVToRGB<GPUDevice, T>::operator()( \ 134 const GPUDevice& d, TTypes<T, 2>::ConstTensor input_data, \ 136 extern template struct HSVToRGB<GPUDevice, T>; 143 RGBToHSVOp<GPUDevice, T>); \ 146 HSVToRGBOp<GPUDevice, T>) [all...] |
data_format_ops.cc | 29 typedef Eigen::GpuDevice GPUDevice; 137 void DataFormatDimMap<GPUDevice, T>::operator()( \ 138 const GPUDevice& d, typename TTypes<T>::ConstFlat x, \ 140 extern template struct DataFormatDimMap<GPUDevice, T>; 148 void DataFormatVecPermute<GPUDevice, T>::operator()( \ 149 const GPUDevice& d, typename TTypes<T>::ConstFlat x, \ 151 extern template struct DataFormatVecPermute<GPUDevice, T>; 162 DataFormatDimMapOp<GPUDevice, T>); 170 DataFormatVecPermuteOp<GPUDevice, T>) [all...] |
softplus_op.cc | 32 typedef Eigen::GpuDevice GPUDevice; 100 void Softplus<GPUDevice, T>::operator()( \ 101 const GPUDevice& d, typename TTypes<T>::ConstTensor features, \ 103 extern template struct Softplus<GPUDevice, T>; \ 106 void SoftplusGrad<GPUDevice, T>::operator()( \ 107 const GPUDevice& d, typename TTypes<T>::ConstTensor gradients, \ 110 extern template struct SoftplusGrad<GPUDevice, T>; 119 SoftplusOp<GPUDevice, type>); \ 122 SoftplusGradOp<GPUDevice, type>) [all...] |
softsign_op.cc | 32 typedef Eigen::GpuDevice GPUDevice; 101 void Softsign<GPUDevice, T>::operator()( \ 102 const GPUDevice& d, typename TTypes<T>::ConstTensor features, \ 104 extern template struct Softsign<GPUDevice, T>; \ 107 void SoftsignGrad<GPUDevice, T>::operator()( \ 108 const GPUDevice& d, typename TTypes<T>::ConstTensor gradients, \ 111 extern template struct SoftsignGrad<GPUDevice, T>; 120 SoftsignOp<GPUDevice, type>); \ 123 SoftsignGradOp<GPUDevice, type>) [all...] |
random_op_gpu.cu.cc | 37 typedef Eigen::GpuDevice GPUDevice; 216 void FillPhiloxRandom<GPUDevice, Distribution>::operator()( 217 OpKernelContext*, const GPUDevice& d, random::PhiloxRandom gen, 233 GPUDevice, random::UniformDistribution<random::PhiloxRandom, Eigen::half> >; 235 GPUDevice, random::UniformDistribution<random::PhiloxRandom, float> >; 237 GPUDevice, random::UniformDistribution<random::PhiloxRandom, double> >; 239 GPUDevice, random::UniformDistribution<random::PhiloxRandom, int32> >; 241 GPUDevice, random::UniformDistribution<random::PhiloxRandom, int64> >; 243 GPUDevice, random::NormalDistribution<random::PhiloxRandom, Eigen::half> > [all...] |
fake_quant_ops.cc | 147 typedef Eigen::GpuDevice GPUDevice; 151 void FakeQuantWithMinMaxArgsFunctor<GPUDevice>::operator()( 152 const GPUDevice& d, typename TTypes<float>::ConstFlat inputs, 155 extern template struct FakeQuantWithMinMaxArgsFunctor<GPUDevice>; 157 FakeQuantWithMinMaxArgsOp<GPUDevice>); 160 void FakeQuantWithMinMaxArgsGradientFunctor<GPUDevice>::operator()( 161 const GPUDevice& d, typename TTypes<float>::ConstFlat gradients, 167 FakeQuantWithMinMaxArgsGradientOp<GPUDevice>); 270 void FakeQuantWithMinMaxVarsFunctor<GPUDevice>::operator() [all...] |
cwise_ops_gpu_common.cu.h | 35 typedef Eigen::GpuDevice GPUDevice; 39 // Partial specialization of UnaryFunctor<Device=GPUDevice, Functor>. 41 struct UnaryFunctor<GPUDevice, Functor> { 42 void operator()(const GPUDevice& d, typename Functor::tout_type out, 48 // Partial specialization of BinaryFunctor<Device=GPUDevice, Functor>. 50 struct BinaryFunctor<GPUDevice, Functor, NDIMS, has_errors> { 51 void operator()(const GPUDevice& d, typename Functor::tout_type out, 58 void Left(const GPUDevice& d, typename Functor::tout_type out, 68 void Right(const GPUDevice& d, typename Functor::tout_type out [all...] |
argmax_op.cc | 41 typedef Eigen::GpuDevice GPUDevice; 156 void ArgMax<GPUDevice, T, Tout>::Reduce##Dims( \ 157 const GPUDevice& d, typename TTypes<T, Dims>::ConstTensor input, \ 160 void ArgMin<GPUDevice, T, Tout>::Reduce##Dims( \ 161 const GPUDevice& d, typename TTypes<T, Dims>::ConstTensor input, \ 177 extern template struct ArgMax<GPUDevice, T, int64>; \ 178 extern template struct ArgMin<GPUDevice, T, int64>; \ 179 extern template struct ArgMax<GPUDevice, T, int32>; \ 180 extern template struct ArgMin<GPUDevice, T, int32> [all...] |
determinant_op_gpu.cu.cc | 31 typedef Eigen::GpuDevice GPUDevice; 124 struct DeterminantFromPivotedLUFunctor<GPUDevice, Scalar> { 125 void operator()(const GPUDevice& device, 139 template struct DeterminantFromPivotedLUFunctor<GPUDevice, float>; 140 template struct DeterminantFromPivotedLUFunctor<GPUDevice, double>; 141 template struct DeterminantFromPivotedLUFunctor<GPUDevice, complex64>; 142 template struct DeterminantFromPivotedLUFunctor<GPUDevice, complex128>; 145 struct LogDeterminantFromPivotedLUFunctor<GPUDevice, Scalar> { 146 void operator()(const GPUDevice& device [all...] |
pooling_ops_common_gpu.h | 41 typedef GPUDevice Device; 56 typedef GPUDevice Device;
|
scatter_functor_gpu.cu.cc | 24 typedef Eigen::GpuDevice GPUDevice; 27 template struct functor::ScatterFunctor<GPUDevice, T, Index, op>;
|
scatter_op_gpu.cu.cc | 24 typedef Eigen::GpuDevice GPUDevice; 28 template struct functor::ScatterFunctor<GPUDevice, T, Index, op>;
|