Searched defs:concatenate_dimension (Results 1 – 5 of 5) sorted by relevance
228 int concatenate_dimension) { in Concat2D()251 int concatenate_dimension) { in Concat3D()286 int concatenate_dimension) { in Concat4D()
566 int64 concatenate_dimension() const { return dimensions(0); } in concatenate_dimension() function
629 int64 concatenate_dimension = hlo->concatenate_dimension(); in HandleConcatenate() local
3910 int64 HloInstruction::concatenate_dimension() const { in concatenate_dimension() function in xla::HloInstruction
1246 int64 concatenate_dimension = concatenate->concatenate_dimension(); in HandleConcatenate() local