Searched defs:set_shapes (Results 1 – 2 of 2) sorted by relevance
191 auto set_shapes = [&a_shapes_and_types, &b_shapes_and_types]( in TEST() local227 auto set_shapes = [&a_shapes_and_types, &b_shapes_and_types]( in TEST() local327 auto set_shapes = [&a_shapes_and_types, &b_shapes_and_types]( in TEST() local
100 void set_shapes(const Shape& on_device_shape) { in set_shapes() function109 void set_shapes(const Shape& on_host_shape, const Shape& on_device_shape) { in set_shapes() function