Searched refs:ReductionAxes (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | reduction_ops.h | 43 typename ReductionAxes, typename Reducer>
46 const ReductionAxes& reduction_axes, const Reducer& reducer) { in operator()
52 typename ReductionAxes, typename Scalar>
53 struct ReduceEigenImpl<Device, OUT_T, IN_T, ReductionAxes,
56 const ReductionAxes& reduction_axes,
68 typename ReductionAxes, typename Scalar>
69 struct ReduceEigenImpl<Device, OUT_T, IN_T, ReductionAxes,
72 const ReductionAxes& reduction_axes,
82 typename ReductionAxes>
83 struct ReduceEigenImpl<Device, OUT_T, IN_T, ReductionAxes,
[all …]
|
| D | reduction_gpu_kernels.cu.h | 852 typename ReductionAxes>
855 const ReductionAxes& reduction_axes, Op op) {
886 template <typename OUT_T, typename IN_T, typename ReductionAxes>
888 const ReductionAxes& reduction_axes,
894 template <typename OUT_T, typename IN_T, typename ReductionAxes>
896 const ReductionAxes& reduction_axes,
898 ReduceImpl<T, Sum<T>, T*, T*, ReductionAxes>(
915 template <typename OUT_T, typename IN_T, typename ReductionAxes>
917 const ReductionAxes& reduction_axes,
923 ReduceImpl<T, Sum<T>, outputIterType, inputIterType, ReductionAxes>(
[all …]
|
| D | l2loss_op_gpu.cu.cc | 54 typedef const Eigen::array<TTypes<float>::Tensor::Index, 1>& ReductionAxes; in Compute() typedef
57 functor::ReduceImpl<T, cub::Sum, T*, inputIterType, ReductionAxes>( in Compute()
|
| D | sparse_xent_op_gpu.cu.cc | 49 typedef const Eigen::array<TTypes<float>::Tensor::Index, 1>& ReductionAxes; in Compute() typedef
52 functor::ReduceImpl<T, cub::Max, T*, const T*, ReductionAxes>( in Compute()
|
| D | bias_op_gpu.cu.cc | 254 typedef const Eigen::array<TTypes<float>::Tensor::Index, 1>& ReductionAxes; in DoRowReduction() typedef
257 functor::ReduceImpl<T, cub::Sum, T*, const T*, ReductionAxes>( in DoRowReduction()
264 typedef const Eigen::array<TTypes<float>::Tensor::Index, 1>& ReductionAxes; in DoColReduction() typedef
267 functor::ReduceImpl<T, cub::Sum, T*, const T*, ReductionAxes>( in DoColReduction()
|
| D | reduction_ops_common.h | 254 template <typename OUT_T, typename IN_T, typename ReductionAxes>
256 const ReductionAxes& reduction_axes,
259 ReduceEigenImpl<Device, OUT_T, IN_T, ReductionAxes, Reducer> reducer_impl;
|
| D | multinomial_op_gpu.cu.cc | 100 typedef const Eigen::array<TTypes<float>::Tensor::Index, 1>& ReductionAxes; in operator ()() typedef
103 functor::ReduceImpl<float, cub::Max, float*, const float*, ReductionAxes>( in operator ()()
|
| D | softmax_op_gpu.cu.cc | 118 typedef const Eigen::array<TTypes<float>::Tensor::Index, 1>& ReductionAxes; in DoRowReduction() typedef
123 functor::ReduceImpl<T, Op, T*, InputIter, ReductionAxes>( in DoRowReduction()
|