Searched refs:tensor_num_elements (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | quantized_mul_op.cc | 191 int32 tensor_offset, int64 tensor_num_elements, in VectorTensorMultiply() argument
193 for (int i = 0; i < tensor_num_elements; ++i) { in VectorTensorMultiply()
204 const quint8* tensor_data, int32 tensor_offset, int64 tensor_num_elements, in VectorTensorMultiply() argument
208 CHECK_EQ(0, tensor_num_elements % vector_num_elements); in VectorTensorMultiply()
209 for (int base_i = 0; base_i < tensor_num_elements; in VectorTensorMultiply()
333 int64 tensor_num_elements; in Compute() local
340 tensor_num_elements = y.NumElements(); in Compute()
347 tensor_num_elements = x.NumElements(); in Compute()
352 tensor_offset, tensor_num_elements, z_data); in Compute()
|
| D | quantized_add_op.cc | 292 float max_tensor, int64 tensor_num_elements, in VectorTensorAddition() argument
294 for (int i = 0; i < tensor_num_elements; ++i) { in VectorTensorAddition()
310 float max_tensor, int64 tensor_num_elements, in VectorTensorAddition() argument
343 for (int64 base_i = 0; base_i < tensor_num_elements; in VectorTensorAddition()
395 float max_tensor, int64 tensor_num_elements, in VectorTensorAddition() argument
422 for (int i = 0; i < tensor_num_elements; ++i) { in VectorTensorAddition()
519 int64 tensor_num_elements; in Compute() local
528 tensor_num_elements = y.NumElements(); in Compute()
537 tensor_num_elements = x.NumElements(); in Compute()
543 tensor_min, tensor_max, tensor_num_elements, min_z_value, max_z_value, in Compute()
|