Home
last modified time | relevance | path

Searched defs:ReductionAxes (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dl2loss_op_gpu.cu.cc53 typedef const Eigen::array<TTypes<float>::Tensor::Index, 1>& ReductionAxes; in Compute() typedef
Dsparse_xent_op_gpu.cu.cc49 typedef const Eigen::array<TTypes<float>::Tensor::Index, 1>& ReductionAxes; in Compute() typedef
Dmultinomial_op_gpu.cu.cc103 typedef const Eigen::array<TTypes<float>::Tensor::Index, 1>& ReductionAxes; in operator ()() typedef
Dbias_op_gpu.cu.cc269 typedef const Eigen::array<TTypes<float>::Tensor::Index, 1>& ReductionAxes; in DoRowReduction() typedef
279 typedef const Eigen::array<TTypes<float>::Tensor::Index, 1>& ReductionAxes; in DoColReduction() typedef
Dsoftmax_op_gpu.cu.cc184 typedef const Eigen::array<TTypes<float>::Tensor::Index, 1>& ReductionAxes; in DoRowReduction() typedef