Searched refs:tensor_num_elements (Results 1 – 5 of 5) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | quantized_mul_op.cc | 192 int32_t tensor_offset, int64_t tensor_num_elements, in VectorTensorMultiply() argument
194 for (int i = 0; i < tensor_num_elements; ++i) { in VectorTensorMultiply()
205 const quint8* tensor_data, int32 tensor_offset, int64 tensor_num_elements, in VectorTensorMultiply() argument
209 CHECK_EQ(0, tensor_num_elements % vector_num_elements); in VectorTensorMultiply()
210 for (int base_i = 0; base_i < tensor_num_elements; in VectorTensorMultiply()
346 int64_t tensor_num_elements; in Compute() local
353 tensor_num_elements = y.NumElements(); in Compute()
360 tensor_num_elements = x.NumElements(); in Compute()
370 tensor_offset, tensor_num_elements, z_data); in Compute()
|
| H A D | quantized_add_op.cc | 293 float max_tensor, int64_t tensor_num_elements, in VectorTensorAddition() argument
295 for (int i = 0; i < tensor_num_elements; ++i) { in VectorTensorAddition()
311 float max_tensor, int64 tensor_num_elements, in VectorTensorAddition() argument
344 for (int64 base_i = 0; base_i < tensor_num_elements; in VectorTensorAddition()
396 float max_tensor, int64_t tensor_num_elements, in VectorTensorAddition() argument
423 for (int i = 0; i < tensor_num_elements; ++i) { in VectorTensorAddition()
538 int64_t tensor_num_elements; in Compute() local
547 tensor_num_elements = y.NumElements(); in Compute()
556 tensor_num_elements = x.NumElements(); in Compute()
564 tensor_min, tensor_max, tensor_num_elements, min_z_value, max_z_value, in Compute()
|
| /aosp_15_r20/external/tensorflow/tensorflow/security/advisory/ |
| H A D | tfsa-2021-017.md | 29 int32 tensor_offset, int64 tensor_num_elements,
31 for (int i = 0; i < tensor_num_elements; ++i) {
|
| H A D | tfsa-2021-037.md | 32 float max_tensor, int64 tensor_num_elements,
34 for (int i = 0; i < tensor_num_elements; ++i) {
|
| /aosp_15_r20/external/ComputeLibrary/tests/validation/ |
| H A D | Validation.cpp | 186 const uint32_t tensor_num_elements = tensor.num_elements(); in validate() local
187 for(uint32_t element_idx = 0; element_idx < tensor_num_elements; ++element_idx) in validate()
|