Home
last modified time | relevance | path

Searched defs:iteration_shape (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/
Dsort_util.cc348 Shape iteration_shape = ShapeUtil::MakeShape(keys_shape.element_type(), in EmitSortInPlace() local
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter_unnested.cc2641 Shape iteration_shape = keys_shape; in EmitSort() local