Searched refs:host_shapes (Results 1 – 3 of 3) sorted by relevance
218 std::vector<Shape> host_shapes(tuple.size()); in GetPythonBufferTree() local219 for (int i = 0; i < host_shapes.size(); ++i) { in GetPythonBufferTree()228 host_shapes[i] = std::move(subtree.shape); in GetPythonBufferTree()230 tree.shape = ShapeUtil::MakeTupleShape(host_shapes); in GetPythonBufferTree()
778 std::vector<const Shape*> host_shapes(4); in BM_DynamicSlice() local790 host_shapes[i] = &shaped_buffers[i].on_host_shape(); in BM_DynamicSlice()801 client->Compile(computation, host_shapes, ExecutableBuildOptions())); in BM_DynamicSlice()
1485 std::vector<Shape> host_shapes; in MakeTupleHelper() local1487 host_shapes.reserve(py_buffers.size()); in MakeTupleHelper()