Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dstable_sort_expander.cc54 sort->sort_dimension() && in ExpandInstruction()
69 if (iota_shape.dimensions(sort->sort_dimension()) > in ExpandInstruction()
76 HloInstruction::CreateIota(iota_shape, sort->sort_dimension())); in ExpandInstruction()
Dhlo_instructions.h502 int64 sort_dimension() const { return dimensions(0); } in sort_dimension() function
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dir_emitter.cc562 keys_shape.layout())[sort->sort_dimension()]; in HandleSort()