Searched defs:body_input_shape (Results 1 – 1 of 1) sorted by relevance
174 xla::Shape body_input_shape = body.xla_input_shapes[0]; in VerifyBodyInputAndOutputShapeMatch() local218 xla::Shape body_input_shape = body.xla_input_shapes[0]; in BuildWrappedBody() local439 xla::Shape body_input_shape = body.xla_input_shapes[0]; in Compile() local