Searched defs:GemmConfig (Results 1 – 2 of 2) sorted by relevance
89 struct GemmConfig { struct114 Status RunGemm(const GemmConfig& config, se::DeviceMemoryBase lhs_buffer, argument
106 struct GemmConfig struct108 GemmMethod method = GemmMethod::DEFAULT;109 std::string filter = "";110 unsigned int inner_block_size = 0;111 unsigned int outer_block_size = 0;112 WeightFormat weight_format = WeightFormat::ANY;114 GemmConfig(GemmMethod method) in GemmConfig() argument118 GemmConfig() in GemmConfig() argument