Searched refs:uniform_shape (Results 1 - 1 of 1) sorted by relevance
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/ | ||
H A D | categorical_op.cc | 66 xla::Shape uniform_shape = variable 70 XlaHelpers::One(builder, input_type(0)), uniform_shape); |
Completed in 137 milliseconds