Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/
Dlayout_util.h60 static Layout GetDefaultLayoutForRank(int64_t rank);
Dlayout_util.cc129 /* static */ Layout LayoutUtil::GetDefaultLayoutForRank(int64_t rank) { in GetDefaultLayoutForRank() function in xla::LayoutUtil
Dliteral_util.h408 values, LayoutUtil::GetDefaultLayoutForRank(values.num_dimensions())); in CreateFromArray()
Dliteral.cc968 output = Relayout(LayoutUtil::GetDefaultLayoutForRank(shape().rank())); in Reshape()
/external/tensorflow/tensorflow/compiler/xla/service/
Dall_to_all_decomposer.cc82 LayoutUtil::GetDefaultLayoutForRank(min_array_rank_); in ExpandInstruction()
Dhlo_evaluator.cc1854 LayoutUtil::GetDefaultLayoutForRank(lengths.size())); in ComputeStrides()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_layout_assignment.cc340 LayoutUtil::GetDefaultLayoutForRank(keys_shape.rank()); in AddBackendConstraints()