Searched defs:vect_size (Results 1 – 5 of 5) sorted by relevance
92 static XlaOp SplitAtDim(XlaOp instr, int64_t dim, int64_t vect_size) { in SplitAtDim()108 static Shape SplitShapeAtDim(Shape shape, int64_t dim, int64_t vect_size) { in SplitShapeAtDim()141 int64_t vect_size) { in RevectorizeInstr()262 HloCustomCallInstruction* conv, int vect_size) { in TryRevectorizeConv()400 HloCustomCallInstruction* conv, int64_t vect_size) { in TryVectorizeConv()
203 int64_t vect_size = input.dimensions(input.layout().minor_to_major(0)); in XlaConvShapesToStreamExecutorLayouts() local227 int64_t vect_size = filter.dimensions(filter.layout().minor_to_major(0)); in XlaConvShapesToStreamExecutorLayouts() local251 int64_t vect_size = output.dimensions(output.layout().minor_to_major(0)); in XlaConvShapesToStreamExecutorLayouts() local
1104 lsc_vector_length(enum lsc_vect_size vect_size) in lsc_vector_length()1121 lsc_vect_size(unsigned vect_size) in lsc_vect_size()
1281 lsc_vector_length(enum lsc_vect_size vect_size) in lsc_vector_length()1298 lsc_vect_size(unsigned vect_size) in lsc_vect_size()
552 absl::optional<DimensionHandle> vect_size, in ShapeFromDimensions()744 absl::optional<DimensionHandle> vect_size; in Conv2DShapeImpl() local