Searched defs:concatenate_dimension (Results 1 – 5 of 5) sorted by relevance
240 int concatenate_dimension) { in Concat2D()263 int concatenate_dimension) { in Concat3D()298 int concatenate_dimension) { in Concat4D()
843 int64_t concatenate_dimension() const override { in concatenate_dimension() function
659 int64_t concatenate_dimension = hlo->concatenate_dimension(); in HandleConcatenate() local
4478 int64_t HloInstruction::concatenate_dimension() const { in concatenate_dimension() function in xla::HloInstruction
1421 int64_t concatenate_dimension = concatenate->concatenate_dimension(); in HandleConcatenate() local