Searched refs:input_element_count (Results 1 – 4 of 4) 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 ()()
237 const auto input_element_count = input_shape.num_elements(); in operator ()() local241 input_element_count >= std::min(num_threads, num_split) * 4096 && in operator ()()242 input_element_count < num_split * 180 * 1024); in operator ()()282 num_split, input_element_count / num_split, range_output_func); in operator ()()
830 const int64_t input_element_count = input.NumElements(); in QuantizedAddUsingEigen() local847 bcast[0] = input_element_count / smaller_input_element_count; in QuantizedAddUsingEigen()876 const int64_t input_element_count = input.NumElements(); in QuantizedAdd() local882 (input_element_count / smaller_input_element_count); in QuantizedAdd()
39 const int64 input_element_count = input.NumElements();42 bcast[0] = input_element_count / smaller_input_element_count;