Searched refs:individual_arg_shapes (Results 1 – 1 of 1) sorted by relevance
96 std::vector<xla::Shape> individual_arg_shapes; in GetXlaInputShapes() local97 individual_arg_shapes.reserve(num_args); in GetXlaInputShapes()99 individual_arg_shapes.emplace_back(); in GetXlaInputShapes()100 xla::Shape& xla_shape = individual_arg_shapes.back(); in GetXlaInputShapes()126 xla::ShapeUtil::MakeTupleShape(individual_arg_shapes)); in GetXlaInputShapes()128 *xla_input_shapes = individual_arg_shapes; in GetXlaInputShapes()