Searched refs:MatMultDims (Results 1 – 1 of 1) sorted by relevance
152 struct MatMultDims { struct in xla::cpu::__anon162db8010111::DotOpEmitter179 MatMultDims GetMatMultDims() const;184 MatMultDims GetBatchMatMultDims() const;381 MatMultDims mat_mult_dims = GetMatMultDims(); in EmitTiledLlvmIrGemm()424 MatMultDims mat_mult_dims = GetMatMultDims(); in EmitTiledLlvmIrGemv()890 MatMultDims mat_mult_dims = GetMatMultDims(); in EmitCallToRuntime()973 MatMultDims mat_mult_dims = GetBatchMatMultDims(); in EmitCallToBatchRuntime()1003 DotOpEmitter::MatMultDims DotOpEmitter::GetMatMultDims() const { in GetMatMultDims()1033 DotOpEmitter::MatMultDims DotOpEmitter::GetBatchMatMultDims() const { in GetBatchMatMultDims()