Lines Matching defs:Vec
74 // it's more convenient to manipulate Vec directly for this module.
75 typedef gtl::InlinedVector<int64, 4> Vec;
85 BCast(const Vec& x, const Vec& y, const bool fewer_dims_optimization = true);
95 const Vec& x_reshape() const { return x_reshape_; }
96 const Vec& x_bcast() const { return x_bcast_; }
97 const Vec& y_reshape() const { return y_reshape_; }
98 const Vec& y_bcast() const { return y_bcast_; }
99 const Vec& result_shape() const { return result_; }
100 const Vec& output_shape() const { return output_; }
101 const Vec& grad_x_reduce_idx() const { return grad_x_reduce_idx_; }
102 const Vec& grad_y_reduce_idx() const { return grad_y_reduce_idx_; }
105 static Vec FromShape(const TensorShape& shape);
106 static TensorShape ToShape(const BCast::Vec& vec);
110 const BCast::Vec& vec) {
111 CHECK_EQ(vec.size(), NDIMS);
113 for (int i = 0; i < NDIMS; ++i) ret[i] = vec[i];
119 Vec x_reshape_;
120 Vec x_bcast_;
121 Vec y_reshape_;
122 Vec y_bcast_;
123 Vec result_;
124 Vec output_;
125 Vec grad_x_reduce_idx_;
126 Vec grad_y_reduce_idx_;
128 static void Reverse(Vec* shape);
