HomeSort by relevance Sort by last modified time
    Searched refs:ReductionHelper (Results 1 - 2 of 2) sorted by null

  /external/tensorflow/tensorflow/core/kernels/
reduction_ops_common.cc 22 TensorShape ReductionHelper::out_reshape() const {
29 TensorShape ReductionHelper::out_shape() const {
35 TensorShape ReductionHelper::shuffled_shape() {
47 gtl::InlinedVector<int32, 8> ReductionHelper::permutation() {
77 Status ReductionHelper::Simplify(const Tensor& data, const Tensor& axis,
reduction_ops_common.h 80 class ReductionHelper {
82 ReductionHelper() : reduce_first_axis_(false) {}
154 ReductionHelper helper;

Completed in 114 milliseconds