Searched refs:device_shapes (Results 1 – 2 of 2) sorted by relevance
720 device_shapes = [array_ops.shape(g) for g, _ in device_grads_and_vars]753 self.all_device_shapes.append(device_shapes)762 device_grads_and_vars, device_shapes, device_sizes) in zip(781 for shape, grad in zip(device_shapes, grads_with_sizes)
1486 std::vector<Shape> device_shapes; in MakeTupleHelper() local1488 device_shapes.reserve(py_buffers.size()); in MakeTupleHelper()1490 device_shapes.push_back(buffer->on_device_shape()); in MakeTupleHelper()1492 Shape on_device_shape = ShapeUtil::MakeTupleShape(device_shapes); in MakeTupleHelper()