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