Home
last modified time | relevance | path

Searched refs:operand_buffers (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dcudnn_conv_runner.cc322 absl::Span<se::DeviceMemoryBase> operand_buffers, in GetCudnnConvParams() argument
346 params.input_buf = operand_buffers[0]; in GetCudnnConvParams()
347 params.filter_buf = operand_buffers[1]; in GetCudnnConvParams()
355 params.filter_buf = operand_buffers[1]; in GetCudnnConvParams()
356 params.output_buf = operand_buffers[0]; in GetCudnnConvParams()
362 params.input_buf = operand_buffers[0]; in GetCudnnConvParams()
364 params.output_buf = operand_buffers[1]; in GetCudnnConvParams()
380 params.input_buf = operand_buffers[0]; in GetCudnnConvParams()
381 params.filter_buf = operand_buffers[1]; in GetCudnnConvParams()
383 params.fusion->bias_buf = operand_buffers[2]; in GetCudnnConvParams()
[all …]
Dcudnn_conv_runner.h54 absl::Span<se::DeviceMemoryBase> operand_buffers,
60 absl::Span<se::DeviceMemoryBase> operand_buffers,
Dcudnn_conv_algorithm_picker.cc211 std::vector<se::DeviceMemoryBase> operand_buffers; in PickBestAlgorithm() local
217 operand_buffers.push_back(buffer); in PickBestAlgorithm()
246 RunCudnnConv(instr, absl::MakeSpan(operand_buffers), result_buffer, in PickBestAlgorithm()
/external/tensorflow/tensorflow/compiler/xla/service/
Dlayout_assignment.cc147 PointsToSet::BufferSet* operand_buffers = in OperandBufferForwarded() local
150 return operand_buffers->count(b) > 0; in OperandBufferForwarded()