Searched defs:uniform_shape (Results 1 – 1 of 1) sorted by relevance
69 xla::Shape uniform_shape; in Compile() local132 virtual xla::XlaOp GetLogUniforms(xla::Shape uniform_shape, in GetLogUniforms()164 xla::XlaOp GetLogUniforms(xla::Shape uniform_shape, xla::PrimitiveType type, in GetLogUniforms()