Searched refs:shaped_buffers (Results 1 – 2 of 2) sorted by relevance
295 std::vector<ShapedBuffer> shaped_buffers; in RunAsync() local297 shaped_buffers.reserve(arguments.size()); in RunAsync()300 shaped_buffers.push_back(MaybeOwningShapeTreeToShapedBuffer( in RunAsync()302 shaped_buffer_ptrs.push_back(&shaped_buffers.back()); in RunAsync()
777 std::vector<ScopedShapedBuffer> shaped_buffers; in BM_DynamicSlice() local784 shaped_buffers.emplace_back( in BM_DynamicSlice()790 host_shapes[i] = &shaped_buffers[i].on_host_shape(); in BM_DynamicSlice()792 stream.get(), start_index_literal, shaped_buffers[i])); in BM_DynamicSlice()809 absl::c_transform(shaped_buffers, std::back_inserter(shaped_buffer_ptrs), in BM_DynamicSlice()