Searched refs:iota_dimensions (Results 1 – 1 of 1) sorted by relevance
106 std::vector<int64> iota_dimensions(expanded_filter_shape.dimensions().begin(), in CreateExpandedFilterMask() local108 xla::Shape iota_shape = xla::ShapeUtil::MakeShape(xla::S32, iota_dimensions); in CreateExpandedFilterMask()110 builder, iota_shape, /*iota_dimension=*/iota_dimensions.size() - 2); in CreateExpandedFilterMask()112 builder, iota_shape, /*iota_dimension=*/iota_dimensions.size() - 1); in CreateExpandedFilterMask()