Home
last modified time | relevance | path

Searched refs:source_slices (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Doutfeed_thunk.cc29 std::vector<ShapedSlice> source_slices) in OutfeedThunk() argument
31 source_slices_(std::move(source_slices)) {} in OutfeedThunk()
Doutfeed_thunk.h35 OutfeedThunk(ThunkInfo thunk_info, std::vector<ShapedSlice> source_slices);
Dir_emitter_unnested.cc3214 std::vector<ShapedSlice> source_slices; in HandleOutfeed() local
3215 source_slices.reserve(outfeed_op.operands().size()); in HandleOutfeed()
3220 source_slices.push_back(ShapedSlice{slice, shape}); in HandleOutfeed()
3224 input.thunk_info, std::move(source_slices))); in HandleOutfeed()