Home
last modified time | relevance | path

Searched defs:uniform_shape (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dcategorical_op.cc69 xla::Shape uniform_shape; in Compile() local
132 virtual xla::XlaOp GetLogUniforms(xla::Shape uniform_shape, in GetLogUniforms()
164 xla::XlaOp GetLogUniforms(xla::Shape uniform_shape, xla::PrimitiveType type, in GetLogUniforms()