Searched refs:input_element_count (Results 1 – 3 of 3) sorted by relevance
171 const auto input_element_count = input_shape.num_elements(); in Compute() local174 input_element_count >= std::max(num_threads, num_split) * 4096 && in Compute()175 input_element_count < num_split * 180 * 1024); in Compute()213 num_split, input_element_count / num_split, range_output_func); in Compute()
229 const auto input_element_count = input_shape.num_elements(); in Compute() local232 input_element_count >= std::max(num_threads, num_split) * 4096 && in Compute()233 input_element_count < num_split * 180 * 1024); in Compute()271 num_split, input_element_count / num_split, range_output_func); in Compute()
823 const int64 input_element_count = input.NumElements(); in QuantizedAddUsingEigen() local840 bcast[0] = input_element_count / smaller_input_element_count; in QuantizedAddUsingEigen()869 const int64 input_element_count = input.NumElements(); in QuantizedAdd() local875 (input_element_count / smaller_input_element_count); in QuantizedAdd()