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()
621 int64 concatenate_dimension() const { return dimensions(0); } in concatenate_dimension() function
630 int64_t concatenate_dimension = hlo->concatenate_dimension(); in HandleConcatenate() local
1301 int64_t concatenate_dimension = concatenate->concatenate_dimension(); in HandleConcatenate() local
4207 int64 HloInstruction::concatenate_dimension() const { in concatenate_dimension() function in xla::HloInstruction