Searched refs:input_element_count (Results 1 – 3 of 3) sorted by relevance
143 const auto input_element_count = input_shape.num_elements(); in operator ()() local146 input_element_count >= std::max(num_threads, num_split) * 4096 && in operator ()()147 input_element_count < num_split * 180 * 1024); in operator ()()190 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::min(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 ()()
832 const int64 input_element_count = input.NumElements(); in QuantizedAddUsingEigen() local849 bcast[0] = input_element_count / smaller_input_element_count; in QuantizedAddUsingEigen()878 const int64 input_element_count = input.NumElements(); in QuantizedAdd() local884 (input_element_count / smaller_input_element_count); in QuantizedAdd()