Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgemm_rewriter.cc63 GemmBackendConfig gemm_config; in HandleDot() local
64 gemm_config.set_alpha_real(1.0); in HandleDot()
65 gemm_config.set_alpha_imag(0.0); in HandleDot()
66 gemm_config.set_beta(0.0); in HandleDot()
67 *gemm_config.mutable_dot_dimension_numbers() = in HandleDot()
69 gemm_config.set_batch_size(batch_size); in HandleDot()
70 TF_RETURN_IF_ERROR(gemm_call->set_backend_config(gemm_config)); in HandleDot()
Dgemm_algorithm_picker.cc232 const HloInstruction* instr, const GemmBackendConfig& gemm_config, in DoGemmAutotune() argument
242 instr->shape(), gemm_config.SerializeAsString()); in DoGemmAutotune()
262 int64 batch_size = gemm_config.batch_size(); in DoGemmAutotune()
286 GemmBackendConfig gemm_config = in RunOnInstruction() local
290 DoGemmAutotune(instr, gemm_config, allocator, stream)); in RunOnInstruction()
294 GemmBackendConfig updated_config = gemm_config; in RunOnInstruction()
299 return updated_config.SerializeAsString() != gemm_config.SerializeAsString(); in RunOnInstruction()
Dgemm_thunk.cc171 Status RunGemm(const GpuGemmConfig &gemm_config, in RunGemm() argument
181 const Shape &output_shape = gemm_config.output_shape; in RunGemm()
182 const Shape &lhs_shape = gemm_config.lhs_shape; in RunGemm()
183 const Shape &rhs_shape = gemm_config.rhs_shape; in RunGemm()
184 const GemmBackendConfig &backend_config = gemm_config.backend_config; in RunGemm()
Dgemm_thunk.h80 const GpuGemmConfig& gemm_config, se::DeviceMemoryBase lhs_buffer,