Searched refs:assn_computation (Results 1 – 2 of 2) sorted by relevance
139 xla::XlaComputation assn_computation = [&] { in Compile() local150 output = xla::Scatter(output, idx, updates, assn_computation, scatter_dnums, in Compile()
282 xla::XlaComputation assn_computation = [&] { in CompileWhereWithPrefixSum() local301 /*update_computation=*/assn_computation, scatter_dnums, in CompileWhereWithPrefixSum()