Searched refs:input_element_count (Results 1 – 3 of 3) sorted by relevance
146 const auto input_element_count = input_shape.num_elements(); in operator ()() local149 input_element_count >= std::max(num_threads, num_split) * 4096 && in operator ()()150 input_element_count < num_split * 180 * 1024); in operator ()()193 num_split, input_element_count / num_split, range_output_func); in operator ()()
205 const auto input_element_count = input_shape.num_elements(); in operator ()() local209 input_element_count >= std::max(num_threads, num_split) * 4096 && in operator ()()210 input_element_count < num_split * 180 * 1024); in operator ()()247 num_split, input_element_count / num_split, range_output_func); in operator ()()
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()