HomeSort by relevance Sort by last modified time
    Searched refs:input_element_count (Results 1 - 3 of 3) sorted by null

  /external/tensorflow/tensorflow/core/kernels/
split_op.cc 146 const auto input_element_count = input_shape.num_elements();
149 input_element_count >= std::max(num_threads, num_split) * 4096 &&
150 input_element_count < num_split * 180 * 1024);
193 num_split, input_element_count / num_split, range_output_func);
split_v_op.cc 205 const auto input_element_count = input_shape.num_elements(); local
209 input_element_count >= std::max(num_threads, num_split) * 4096 &&
210 input_element_count < num_split * 180 * 1024);
247 num_split, input_element_count / num_split, range_output_func);
quantization_utils.h     [all...]

Completed in 57 milliseconds