Home
last modified time | relevance | path

Searched refs:xla_scalar_shape (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2xla/lib/
Dscatter.cc178 auto xla_scalar_shape = in XlaScatter() local
180 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()