OSSForks/tensorflow
1
0
Fork 0
mirror of https://github.com/zebrajr/tensorflow.git synced 2025-12-06 00:19:58 +01:00
Code Issues Actions 5 Packages Projects Releases Wiki Activity

Minor cleanup in YnnDimensions.

Browse Source 
PiperOrigin-RevId: 825955545
This commit is contained in:
Alexander Shaposhnikov 2025-10-30 03:27:00 -07:00 committed by TensorFlower Gardener
parent cfa7953589
commit d29d0f8635
1 changed files with 2 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
third_party/xla/xla/backends/cpu/ynn_emitter.cc vendored
View File

@ -57,11 +57,8 @@ using TensorIdMap = absl::flat_hash_map<const HloInstruction*, uint32_t>;
//===----------------------------------------------------------------------===//
static std::vector<size_t> YnnDimensions(const Shape& shape) {
std::vector<size_t> dims;
for (auto& dim : shape.dimensions()) {
dims.push_back(dim);
}
return dims;
absl::Span<const int64_t> dims = shape.dimensions();
return {dims.begin(), dims.end()};
}
//===----------------------------------------------------------------------===//