Searched defs:cond_input_shape (Results 1 – 1 of 1) sorted by relevance
204 xla::Shape cond_input_shape = cond.xla_input_shapes[0]; in BuildWrappedCond() local456 xla::Shape cond_input_shape = cond.xla_input_shapes[0]; in Compile() local