Searched refs:num_elems_in_a (Results 1 – 1 of 1) sorted by relevance
1244 int64 num_elems_in_a = in PredictSparseTensorDenseMatMul() local1253 const int64 op_count = kOpsPerMac * num_elems_in_a * n_dim; in PredictSparseTensorDenseMatMul()1262 num_elems_in_a * n_dim * DataTypeSize(BaseType(b_matrix.dtype())); in PredictSparseTensorDenseMatMul()