Searched refs:num_value_elements (Results 1 – 2 of 2) sorted by relevance
31 int num_value_elements = data.shape().num_elements() / num_batch_elements;
149 int num_value_elements = data.shape().num_elements() / num_batch_elements; in Compute() local157 for (int v = 0; v < num_value_elements; ++v) { in Compute()