Searched refs:sizes_ (Results 1 – 6 of 6) sorted by relevance
105 : sizes_(sizes.begin(), sizes.end()), values_(new T[num_elements()]) {252 : sizes_(other.sizes_), values_(new T[num_elements()]) {258 sizes_ = other.sizes_;313 std::vector<int64> index(sizes_.size());321 std::vector<int64> index(sizes_.size());331 std::vector<int64> index(sizes_.size());345 std::vector<int64> index(sizes_.size());406 CHECK(n < sizes_.size());407 return sizes_[n];411 const std::vector<int64>& dimensions() const { return sizes_; }[all …]
159 if (sizes_) delete[] sizes_; in ~MklShape()185 sizes_ = new size_t[dimension]; in SetTfLayout()189 sizes_[ii] = sizes[ii]; in SetTfLayout()242 const size_t* GetSizes() const { return sizes_; } in GetSizes()243 int64 dim_size(int index) const { return sizes_[index]; } in dim_size()245 return sizes_[tf_to_mkl_dim_map_[index]]; in tf_dim_size()355 sizes_ = new size_t[dimension_]; in DeSerializeMklShape()359 sizes_[i] = in DeSerializeMklShape()384 sizes_[i]; in SerializeMklShape()407 size_t* sizes_ = nullptr; // Required by MKL for conversions variable[all …]
64 sizes_(sizes) {} in HloRematerialization()110 RematerializationSizes* sizes_; variable
1291 if (sizes_ != nullptr) { in Run()1292 sizes_->before_bytes = before_peak_memory; in Run()1293 sizes_->after_bytes = current_peak_memory; in Run()
709 sizes_.emplace_back(row.size); in Insert()716 const std::deque<int64_t>& sizes() const { return sizes_; } in sizes()723 std::deque<int64_t> sizes_; variable
2214 size_t<3> sizes_; member in cl::NDRange2225 sizes_.push_back(size0); in NDRange()2231 sizes_.push_back(size0); in NDRange()2232 sizes_.push_back(size1); in NDRange()2238 sizes_.push_back(size0); in NDRange()2239 sizes_.push_back(size1); in NDRange()2240 sizes_.push_back(size2); in NDRange()2243 operator const ::size_t*() const { return (const ::size_t*) sizes_; } in operator const::size_t*()