Home
last modified time | relevance | path

Searched refs:ShapedSlice (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Doutfeed_thunk.h34 OutfeedThunk(ThunkInfo thunk_info, std::vector<ShapedSlice> source_slices);
42 const std::vector<ShapedSlice> source_slices_;
Dinfeed_thunk.h35 InfeedThunk(ThunkInfo thunk_info, std::vector<ShapedSlice> dest_slices);
43 const std::vector<ShapedSlice> dest_slices_;
Dinfeed_thunk.cc30 std::vector<ShapedSlice> dest_slices) in InfeedThunk()
Doutfeed_thunk.cc28 std::vector<ShapedSlice> source_slices) in OutfeedThunk()
Dthunk.h147 struct ShapedSlice { struct
Dir_emitter_unnested.h781 StatusOr<std::vector<ShapedSlice>> GetShapedSlices(
Dir_emitter_unnested.cc2912 StatusOr<std::vector<ShapedSlice>> IrEmitterUnnested::GetShapedSlices( in GetShapedSlices()
2914 std::vector<ShapedSlice> shaped_slices; in GetShapedSlices()
2918 shaped_slices.push_back(ShapedSlice{slice, GetShape(opnd)}); in GetShapedSlices()