Searched defs:k_dim (Results 1 – 2 of 2) sorted by relevance
1332 DimensionHandle k_dim; in TopKShapeFn() local
1069 double m_dim, n_dim, k_dim, k_dim_b = 0; in CountMatMulOperations() local1197 TensorShapeProto_Dim k_dim; in GenerateBatchMatmulContextFromEinsum() local