Home
last modified time | relevance | path

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

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