Searched refs:vector_num_elements (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | quantized_mul_op.cc | 190 int64 vector_num_elements, const T* tensor_data, in VectorTensorMultiply() argument
194 const int64 vector_i = i % vector_num_elements; in VectorTensorMultiply()
203 const quint8* vector_data, int32 vector_offset, int64 vector_num_elements, in VectorTensorMultiply() argument
208 CHECK_EQ(0, tensor_num_elements % vector_num_elements); in VectorTensorMultiply()
210 base_i += vector_num_elements) { in VectorTensorMultiply()
212 const int end_i = base_i + vector_num_elements; in VectorTensorMultiply()
215 for (vector_i = 0; vector_i < (vector_num_elements - 15); in VectorTensorMultiply()
330 int64 vector_num_elements; in Compute() local
337 vector_num_elements = x.NumElements(); in Compute()
344 vector_num_elements = y.NumElements(); in Compute()
[all …]
|
| D | quantized_add_op.cc | 290 float max_vector, int64 vector_num_elements, in VectorTensorAddition() argument
295 const int64 vector_i = i % vector_num_elements; in VectorTensorAddition()
308 float max_vector, int64 vector_num_elements, in VectorTensorAddition() argument
344 base_i += vector_num_elements) { in VectorTensorAddition()
347 for (; vector_i < (vector_num_elements - 7); vector_i += 8, i += 8) { in VectorTensorAddition()
363 for (; vector_i < vector_num_elements; ++vector_i, ++i) { in VectorTensorAddition()
393 float max_vector, int64 vector_num_elements, in VectorTensorAddition() argument
423 const int64 vector_i = i % vector_num_elements; in VectorTensorAddition()
515 int64 vector_num_elements; in Compute() local
524 vector_num_elements = x.NumElements(); in Compute()
[all …]
|