Searched refs:gemm_config (Results 1 – 4 of 4) sorted by relevance
63 GemmBackendConfig gemm_config; in HandleDot() local64 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()
232 const HloInstruction* instr, const GemmBackendConfig& gemm_config, in DoGemmAutotune() argument242 instr->shape(), gemm_config.SerializeAsString()); in DoGemmAutotune()262 int64 batch_size = gemm_config.batch_size(); in DoGemmAutotune()286 GemmBackendConfig gemm_config = in RunOnInstruction() local290 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()
171 Status RunGemm(const GpuGemmConfig &gemm_config, in RunGemm() argument181 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()
80 const GpuGemmConfig& gemm_config, se::DeviceMemoryBase lhs_buffer,