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
|