Searched refs:ReorderDims (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/stream_executor/ |
| D | dnn.cc | 401 std::vector<int64_t> ReorderDims(const std::vector<int64_t>& input, in ReorderDims() function
422 std::vector<int64_t> ReorderDims(const std::vector<int64_t>& input, in ReorderDims() function
478 return ReorderDims(bdyx_dims, DataLayout::kBatchDepthYX, layout); in full_dims()
489 return ReorderDims(phys_strides, this->layout(), layout); in full_strides()
499 return dnn::ReorderDims(bdyx_dims, dnn::DataLayout::kBatchDepthYX, layout); in vectorized_dims()
511 return ReorderDims(phys_strides, this->layout(), layout); in vectorized_strides()
695 return ReorderDims(oiyx_dims, FilterLayout::kOutputInputYX, layout); in full_dims()
706 return ReorderDims(phys_strides, this->layout(), layout); in full_strides()
715 return ReorderDims(oiyx_dims, FilterLayout::kOutputInputYX, layout); in vectorized_dims()
727 return ReorderDims(phys_strides, this->layout(), layout); in vectorized_strides()
|
| D | dnn.h | 66 std::vector<int64_t> ReorderDims(const std::vector<int64_t>& input,
|