Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dconvolution_thunk.cc35 std::vector<BufferAllocation::Slice> operand_slices, in ConvolutionThunk() argument
38 operand_buffers_(std::move(operand_slices)), in ConvolutionThunk()
Dconvolution_thunk.h46 std::vector<BufferAllocation::Slice> operand_slices,
Dir_emitter_unnested.cc1119 std::vector<BufferAllocation::Slice> operand_slices; in EmitConvolutionThunkFromMlir() local
1121 operand_slices.reserve(num_operands - 2); in EmitConvolutionThunkFromMlir()
1124 operand_slices.push_back(slice); in EmitConvolutionThunkFromMlir()
1218 input.thunk_info, std::move(config), std::move(operand_slices), in EmitConvolutionThunkFromMlir()