Searched refs:get_i64_array (Results 1 – 2 of 2) sorted by relevance
270 auto get_i64_array = [](absl::Span<const int64_t> container) { in CreateGatherDimensionNumbers() local274 builder.getContext(), get_i64_array(input.offset_dims()), in CreateGatherDimensionNumbers()275 get_i64_array(input.collapsed_slice_dims()), in CreateGatherDimensionNumbers()276 get_i64_array(input.start_index_map()), input.index_vector_dim()); in CreateGatherDimensionNumbers()
621 auto get_i64_array = [](absl::Span<const int64_t> container) { in GetScatterDimensionNumbers() local626 context, get_i64_array(xla_scatter_dim.update_window_dims()), in GetScatterDimensionNumbers()627 get_i64_array(xla_scatter_dim.inserted_window_dims()), in GetScatterDimensionNumbers()628 get_i64_array(xla_scatter_dim.scatter_dims_to_operand_dims()), in GetScatterDimensionNumbers()