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

  /external/tensorflow/tensorflow/core/kernels/
quantized_mul_op.cc 191 int64 vector_num_elements, const T* tensor_data,
195 const int64 vector_i = i % vector_num_elements;
204 const quint8* vector_data, int32 vector_offset, int64 vector_num_elements,
209 CHECK_EQ(0, tensor_num_elements % vector_num_elements);
211 base_i += vector_num_elements) {
213 const int end_i = base_i + vector_num_elements;
216 for (vector_i = 0; vector_i < (vector_num_elements - 15);
331 int64 vector_num_elements; variable
338 vector_num_elements = x.NumElements();
345 vector_num_elements = y.NumElements()
    [all...]
quantized_add_op.cc 291 float max_vector, int64 vector_num_elements,
296 const int64 vector_i = i % vector_num_elements;
309 float max_vector, int64 vector_num_elements,
345 base_i += vector_num_elements) {
348 for (; vector_i < (vector_num_elements - 7); vector_i += 8, i += 8) {
364 for (; vector_i < vector_num_elements; ++vector_i, ++i) {
394 float max_vector, int64 vector_num_elements,
424 const int64 vector_i = i % vector_num_elements;
516 int64 vector_num_elements; variable
525 vector_num_elements = x.NumElements()
    [all...]

Completed in 83 milliseconds