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