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

  /external/tensorflow/tensorflow/core/kernels/
reduction_ops_common.cc 36 const int dims = data_reshape_.size();
39 shape.AddDim(data_reshape_[i]);
42 shape.AddDim(data_reshape_[i]);
48 const int dims = data_reshape_.size();
125 data_reshape_.push_back(data.dim_size(dim_index));
134 data_reshape_.push_back(size);
137 data_reshape_.back() *= size;
141 // are reduced), data_reshape_[1, 3, 5, ...] is out_reshape_,
142 // otherwise, data_reshape_[0, 2, 4, ...] is.
143 for (size_t i = reduce_first_axis_ ? 1 : 0; i < data_reshape_.size()
    [all...]
reduction_ops_common.h 98 int ndims() const { return data_reshape_.size(); }
112 return data.shaped<T, N>(data_reshape_);
118 for (auto s : data_reshape_) shape.AddDim(s);
130 gtl::InlinedVector<int64, 4> data_reshape_; // Reshape data before reduction. member in class:tensorflow::ReductionHelper

Completed in 204 milliseconds