Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/
Dsparse_xent_op_gpu.cu.cc49 typedef const Eigen::array<TTypes<float>::Tensor::Index, 1>& ReductionAxes; in Compute() typedef
Dl2loss_op_gpu.cu.cc54 typedef const Eigen::array<TTypes<float>::Tensor::Index, 1>& ReductionAxes; in Compute() typedef
Dmultinomial_op_gpu.cu.cc100 typedef const Eigen::array<TTypes<float>::Tensor::Index, 1>& ReductionAxes; in operator ()() typedef
Dbias_op_gpu.cu.cc254 typedef const Eigen::array<TTypes<float>::Tensor::Index, 1>& ReductionAxes; in DoRowReduction() typedef
264 typedef const Eigen::array<TTypes<float>::Tensor::Index, 1>& ReductionAxes; in DoColReduction() typedef
Dsoftmax_op_gpu.cu.cc118 typedef const Eigen::array<TTypes<float>::Tensor::Index, 1>& ReductionAxes; in DoRowReduction() typedef