Searched refs:scatter_dnums (Results 1 – 2 of 2) sorted by relevance
100 xla::ScatterDimensionNumbers scatter_dnums; in Compile() local101 scatter_dnums.set_index_vector_dim(1); in Compile()102 scatter_dnums.add_inserted_window_dims(0); in Compile()103 scatter_dnums.add_scatter_dims_to_operand_dims(0); in Compile()107 scatter_dnums.add_inserted_window_dims(1); in Compile()108 scatter_dnums.add_scatter_dims_to_operand_dims(1); in Compile()150 output = xla::Scatter(output, idx, updates, assn_computation, scatter_dnums, in Compile()
292 xla::ScatterDimensionNumbers scatter_dnums; in CompileWhereWithPrefixSum() local293 scatter_dnums.set_index_vector_dim(1); in CompileWhereWithPrefixSum()294 scatter_dnums.add_inserted_window_dims(0); in CompileWhereWithPrefixSum()295 scatter_dnums.add_scatter_dims_to_operand_dims(0); in CompileWhereWithPrefixSum()296 scatter_dnums.add_update_window_dims(1); in CompileWhereWithPrefixSum()301 /*update_computation=*/assn_computation, scatter_dnums, in CompileWhereWithPrefixSum()