Searched refs:dot_info_ (Results 1 – 1 of 1) sorted by relevance
192 DotInfo dot_info_; member in xla::cpu::__anond13f39450111::DotOpEmitter214 : dot_info_(std::move(dot_info)), in DotOpEmitter()226 PrimitiveType primitive_type = dot_info_.result_shape.element_type(); in EmitTiledLlvmIrGemm()264 PrimitiveType primitive_type = dot_info_.result_shape.element_type(); in EmitTiledLlvmIrGemv()387 switch (GetDotImplementationStrategy(hlo_module_config_, dot_info_, in Emit()411 const DotDimensionNumbers& dim_nums = dot_info_.dim_nums; in EmitNaiveLlvmIrGemm()685 CHECK_EQ(dot_info_.result_shape.dimensions_size(), 2); in GetMatMultDims()689 const DotDimensionNumbers& dim_nums = dot_info_.dim_nums; in GetMatMultDims()