Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/
H A Dlayout_util.cc108 Layout CreateDefaultLayoutForRank(int64_t rank) { in CreateDefaultLayoutForRank() function
126 return CreateDefaultLayoutForRank(shape.dimensions_size()); in GetDefaultLayoutForShape()
130 return CreateDefaultLayoutForRank(rank); in GetDefaultLayoutForRank()
134 return CreateDefaultLayoutForRank(2); in GetDefaultLayoutForR2()
138 return CreateDefaultLayoutForRank(3); in GetDefaultLayoutForR3()
142 return CreateDefaultLayoutForRank(4); in GetDefaultLayoutForR4()