Searched refs:combiner_computation (Results 1 – 1 of 1) sorted by relevance
175 xla::XlaComputation combiner_computation; in XlaScatter() local185 combiner_computation = cb.Build().ConsumeValueOrDie(); in XlaScatter()202 return xla::Scatter(buffer, indices, new_updates, combiner_computation, in XlaScatter()