Searched refs:xla_scalar_shape (Results 1 – 1 of 1) sorted by relevance
178 auto xla_scalar_shape = in XlaScatter() local180 auto p0 = xla::Parameter(&cb, 0, xla_scalar_shape, "p0"); in XlaScatter()181 auto p1 = xla::Parameter(&cb, 1, xla_scalar_shape, "p1"); in XlaScatter()