Home
last modified time | relevance | path

Searched refs:shaped_buffers (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/client/
Dlocal_client.cc288 std::vector<ShapedBuffer> shaped_buffers; in RunAsync() local
290 shaped_buffers.reserve(arguments.size()); in RunAsync()
293 shaped_buffers.push_back(MaybeOwningShapeTreeToShapedBuffer( in RunAsync()
296 shaped_buffer_ptrs.push_back(&shaped_buffers.back()); in RunAsync()
/external/tensorflow/tensorflow/compiler/xla/tests/
Ddynamic_ops_test.cc760 std::vector<ScopedShapedBuffer> shaped_buffers; in BM_DynamicSlice() local
767 shaped_buffers.emplace_back( in BM_DynamicSlice()
773 host_shapes[i] = &shaped_buffers[i].on_host_shape(); in BM_DynamicSlice()
775 stream.get(), start_index_literal, shaped_buffers[i])); in BM_DynamicSlice()
792 absl::c_transform(shaped_buffers, std::back_inserter(shaped_buffer_ptrs), in BM_DynamicSlice()