Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_fusible.cc70 std::vector<int64> ExtractRelativeOrderOfNontrivialDims(const Shape& shape) { in ExtractRelativeOrderOfNontrivialDims() function
102 max_rank_order = ExtractRelativeOrderOfNontrivialDims(param->shape()); in LayoutsAreReduceInputFusionFriendly()
108 ExtractRelativeOrderOfNontrivialDims(param->shape()) == in LayoutsAreReduceInputFusionFriendly()