Searched defs:concatenate_dimension (Results 1 – 4 of 4) sorted by relevance
236 int concatenate_dimension) { in Concat2D()259 int concatenate_dimension) { in Concat3D()294 int concatenate_dimension) { in Concat4D()
433 int64 concatenate_dimension() const { return dimensions(0); } in concatenate_dimension() function
678 int64 concatenate_dimension = concatenate->concatenate_dimension(); in HandleConcatenate() local
3234 int64 HloInstruction::concatenate_dimension() const { in concatenate_dimension() function in xla::HloInstruction