Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dspmd_partitioner_util.h61 inline HloInstruction* CreateFirstWithType(PrimitiveType type, SpmdBuilder* b) { in CreateFirstWithType() function
Dspmd_partitioner.cc1672 CreateFirstWithType(element_type, &b_)); in HandleSort()
1835 CreateFirstWithType(element_type, &b_)); in HandleCustomCall()