| /external/tensorflow/tensorflow/core/kernels/ |
| fake_quant_ops_gpu.cu.cc | 25 typedef Eigen::GpuDevice GPUDevice;
28 template struct FakeQuantWithMinMaxArgsFunctor<GPUDevice>;
29 template struct FakeQuantWithMinMaxArgsGradientFunctor<GPUDevice>;
30 template struct FakeQuantWithMinMaxVarsFunctor<GPUDevice>;
31 template struct FakeQuantWithMinMaxVarsGradientFunctor<GPUDevice>;
32 template struct FakeQuantWithMinMaxVarsPerChannelFunctor<GPUDevice>;
33 template struct FakeQuantWithMinMaxVarsPerChannelGradientFunctor<GPUDevice>;
|
| cwise_op_gpu_logical_and.cu.cc | 22 template struct BinaryFunctor<GPUDevice, logical_and, 1>;
23 template struct BinaryFunctor<GPUDevice, logical_and, 2>;
24 template struct BinaryFunctor<GPUDevice, logical_and, 3>;
25 template struct BinaryFunctor<GPUDevice, logical_and, 4>;
26 template struct BinaryFunctor<GPUDevice, logical_and, 5>;
|
| cwise_op_gpu_logical_or.cu.cc | 22 template struct BinaryFunctor<GPUDevice, logical_or, 1>;
23 template struct BinaryFunctor<GPUDevice, logical_or, 2>;
24 template struct BinaryFunctor<GPUDevice, logical_or, 3>;
25 template struct BinaryFunctor<GPUDevice, logical_or, 4>;
26 template struct BinaryFunctor<GPUDevice, logical_or, 5>;
|
| adjust_contrast_op_gpu.cu.cc | 26 typedef Eigen::GpuDevice GPUDevice;
29 template struct functor::AdjustContrastv2<GPUDevice>;
32 template struct functor::AdjustContrast<GPUDevice, uint8>;
33 template struct functor::AdjustContrast<GPUDevice, int8>;
34 template struct functor::AdjustContrast<GPUDevice, int16>;
35 template struct functor::AdjustContrast<GPUDevice, int32>;
36 template struct functor::AdjustContrast<GPUDevice, int64>;
37 template struct functor::AdjustContrast<GPUDevice, float>;
38 template struct functor::AdjustContrast<GPUDevice, double>
[all...] |
| strided_slice_op_gpu.cu.cc | 28 typedef Eigen::GpuDevice GPUDevice;
31 template struct functor::StridedSlice<GPUDevice, T, 1>; \
32 template struct functor::StridedSlice<GPUDevice, T, 2>; \
33 template struct functor::StridedSlice<GPUDevice, T, 3>; \
34 template struct functor::StridedSlice<GPUDevice, T, 4>; \
35 template struct functor::StridedSlice<GPUDevice, T, 5>; \
36 template struct functor::StridedSlice<GPUDevice, T, 6>; \
37 template struct functor::StridedSlice<GPUDevice, T, 7>; \
38 template struct functor::StridedSliceGrad<GPUDevice, T, 1>;
[all...] |
| relu_op_gpu.cu.cc | 29 typedef Eigen::GpuDevice GPUDevice;
33 template struct functor::Relu<GPUDevice, T>; \
34 template struct functor::ReluGrad<GPUDevice, T>; \
35 template struct functor::Relu6<GPUDevice, T>; \
36 template struct functor::Relu6Grad<GPUDevice, T>; \
37 template struct functor::Elu<GPUDevice, T>; \
38 template struct functor::EluGrad<GPUDevice, T>; \
39 template struct functor::Selu<GPUDevice, T>; \
40 template struct functor::SeluGrad<GPUDevice, T>
[all...] |
| argmax_op_gpu.cu.cc | 25 typedef Eigen::GpuDevice GPUDevice;
28 template struct functor::ArgMax<GPUDevice, T, int64>; \
29 template struct functor::ArgMin<GPUDevice, T, int64>; \
30 template struct functor::ArgMax<GPUDevice, T, int32>; \
31 template struct functor::ArgMin<GPUDevice, T, int32>;
|
| batch_norm_op_gpu.cu.cc | 26 typedef Eigen::GpuDevice GPUDevice;
27 template struct functor::BatchNorm<GPUDevice, float>;
28 template struct functor::BatchNorm<GPUDevice, Eigen::half>;
29 template struct functor::BatchNormGrad<GPUDevice, float>;
30 template struct functor::BatchNormGrad<GPUDevice, Eigen::half>;
|
| conv_ops_gpu_2.cu.cc | 27 typedef Eigen::GpuDevice GPUDevice;
28 template struct functor::InflatePadAndShuffle<GPUDevice, float, 4, int>;
29 template struct functor::InflatePadAndShuffle<GPUDevice, float, 4,
31 template struct functor::InflatePadAndShuffle<GPUDevice, Eigen::half, 4, int>;
32 template struct functor::InflatePadAndShuffle<GPUDevice, Eigen::half, 4,
|
| data_format_ops_gpu.cu.cc | 25 typedef Eigen::GpuDevice GPUDevice;
26 template struct functor::DataFormatDimMap<GPUDevice, int32>;
27 template struct functor::DataFormatDimMap<GPUDevice, int64>;
28 template struct functor::DataFormatVecPermute<GPUDevice, int32>;
29 template struct functor::DataFormatVecPermute<GPUDevice, int64>;
|
| colorspace_op_gpu.cu.cc | 25 typedef Eigen::GpuDevice GPUDevice;
28 template class functor::RGBToHSV<GPUDevice, T>; \
29 template class functor::HSVToRGB<GPUDevice, T>;
|
| softplus_op_gpu.cu.cc | 29 typedef Eigen::GpuDevice GPUDevice;
33 template struct functor::Softplus<GPUDevice, T>; \
34 template struct functor::SoftplusGrad<GPUDevice, T>;
|
| softsign_op_gpu.cu.cc | 29 typedef Eigen::GpuDevice GPUDevice;
33 template struct functor::Softsign<GPUDevice, T>; \
34 template struct functor::SoftsignGrad<GPUDevice, T>;
|
| pad_op_gpu.cu.cc | 26 typedef Eigen::GpuDevice GPUDevice;
30 template struct functor::Pad<GPUDevice, T, Tpadding, 0>; \
31 template struct functor::Pad<GPUDevice, T, Tpadding, 1>; \
32 template struct functor::Pad<GPUDevice, T, Tpadding, 2>; \
33 template struct functor::Pad<GPUDevice, T, Tpadding, 3>; \
34 template struct functor::Pad<GPUDevice, T, Tpadding, 4>; \
35 template struct functor::Pad<GPUDevice, T, Tpadding, 5>; \
36 template struct functor::Pad<GPUDevice, T, Tpadding, 6>;
|
| slice_op_gpu.cu.cc | 28 typedef Eigen::GpuDevice GPUDevice;
31 template struct functor::Slice<GPUDevice, T, 1>; \
32 template struct functor::Slice<GPUDevice, T, 2>; \
33 template struct functor::Slice<GPUDevice, T, 3>; \
34 template struct functor::Slice<GPUDevice, T, 4>; \
35 template struct functor::Slice<GPUDevice, T, 5>; \
36 template struct functor::Slice<GPUDevice, T, 6>; \
37 template struct functor::Slice<GPUDevice, T, 7>;
|
| cwise_op_gpu_logical_not.cu.cc | 22 template struct UnaryFunctor<GPUDevice, logical_not>;
|
| quantize_and_dequantize_op_gpu.cu.cc | 27 typedef Eigen::GpuDevice GPUDevice;
31 struct QuantizeAndDequantizeOneScaleFunctor<GPUDevice, T> {
32 void operator()(const GPUDevice& d, typename TTypes<T>::ConstVec input,
36 QuantizeAndDequantizeOneScaleImpl<GPUDevice, T>::Compute(
44 template struct functor::QuantizeAndDequantizeOneScaleFunctor<GPUDevice, float>;
45 template struct functor::QuantizeAndDequantizeOneScaleFunctor<GPUDevice,
|
| adjust_hue_op.h | 26 typedef Eigen::GpuDevice GPUDevice;
31 void operator()(GPUDevice* device, const int64 number_of_elements,
|
| adjust_saturation_op.h | 26 typedef Eigen::GpuDevice GPUDevice;
31 void operator()(GPUDevice* device, const int64 number_of_elements,
|
| cross_op_gpu.cu.cc | 25 typedef Eigen::GpuDevice GPUDevice;
28 template class functor::Cross<GPUDevice, type>;
|
| extract_image_patches_op_gpu.cu.cc | 25 typedef Eigen::GpuDevice GPUDevice;
29 #define REGISTER(T) template struct ExtractImagePatchesForward<GPUDevice, T>;
|
| snapshot_op_gpu.cu.cc | 25 typedef Eigen::GpuDevice GPUDevice;
30 SnapshotOp<GPUDevice, TYPE>);
|
| /external/tensorflow/tensorflow/contrib/image/kernels/ |
| image_ops_gpu.cu.cc | 30 typedef Eigen::GpuDevice GPUDevice;
32 template class FillProjectiveTransform<GPUDevice, uint8>;
33 template class FillProjectiveTransform<GPUDevice, int32>;
34 template class FillProjectiveTransform<GPUDevice, int64>;
35 template class FillProjectiveTransform<GPUDevice, float>;
36 template class FillProjectiveTransform<GPUDevice, double>;
|
| /external/tensorflow/tensorflow/contrib/rnn/kernels/ |
| gru_ops_gpu.cu.cc | 24 typedef Eigen::GpuDevice GPUDevice;
27 template struct GRUBlockCellFprop<GPUDevice, T, true>; \
28 template struct GRUBlockCellBprop<GPUDevice, T, true>;
|
| /external/tensorflow/tensorflow/contrib/framework/kernels/ |
| zero_initializer_op_gpu.cu.cc | 27 using GPUDevice = Eigen::GpuDevice;
29 #define DEFINE_GPU_SPECS(T) template struct TensorSetZero<GPUDevice, T>;
|
