Home
last modified time | relevance | path

Searched refs:phys_dims (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/stream_executor/
Ddnn.cc483 std::vector<int64_t> phys_dims = full_dims(this->layout()); in full_strides() local
484 std::vector<int64_t> phys_strides(phys_dims.size()); in full_strides()
487 phys_strides[i] = phys_strides[i + 1] * phys_dims[i + 1]; in full_strides()
504 std::vector<int64_t> phys_dims = in vectorized_strides() local
506 std::vector<int64_t> phys_strides(phys_dims.size()); in vectorized_strides()
507 phys_strides[phys_dims.size() - 1] = 1; in vectorized_strides()
508 for (int i = phys_dims.size() - 2; i >= 0; i--) { in vectorized_strides()
509 phys_strides[i] = phys_strides[i + 1] * phys_dims[i + 1]; in vectorized_strides()
700 std::vector<int64_t> phys_dims = full_dims(this->layout()); in full_strides() local
701 std::vector<int64_t> phys_strides(phys_dims.size()); in full_strides()
[all …]