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.cc34 std::vector<BufferAllocation::Slice> operand_slices, in ConvolutionThunk() argument
39 operand_buffers_(std::move(operand_slices)), in ConvolutionThunk()
Dconvolution_thunk.h47 std::vector<BufferAllocation::Slice> operand_slices,
Dir_emitter_unnested.cc469 std::vector<BufferAllocation::Slice> operand_slices; in HandleCustomCall() local
470 operand_slices.reserve(custom_call->operand_count()); in HandleCustomCall()
472 operand_slices.push_back(GetAllocationSlice(*operand)); in HandleCustomCall()
479 Cast<HloCustomCallInstruction>(custom_call), std::move(operand_slices), in HandleCustomCall()