Home
last modified time | relevance | path

Searched refs:input_element_count (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dsplit_op.cc171 const auto input_element_count = input_shape.num_elements(); in Compute() local
174 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()
Dsplit_v_op.cc229 const auto input_element_count = input_shape.num_elements(); in Compute() local
232 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()
Dquantization_utils.h823 const int64 input_element_count = input.NumElements(); in QuantizedAddUsingEigen() local
840 bcast[0] = input_element_count / smaller_input_element_count; in QuantizedAddUsingEigen()
869 const int64 input_element_count = input.NumElements(); in QuantizedAdd() local
875 (input_element_count / smaller_input_element_count); in QuantizedAdd()