Searched defs: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()
204 const quint8* tensor_data, int32 tensor_offset, int64 tensor_num_elements, in VectorTensorMultiply()
333 int64 tensor_num_elements; in Compute() local
|
| D | quantized_add_op.cc | 292 float max_tensor, int64 tensor_num_elements, in VectorTensorAddition()
310 float max_tensor, int64 tensor_num_elements, in VectorTensorAddition()
395 float max_tensor, int64 tensor_num_elements, in VectorTensorAddition()
519 int64 tensor_num_elements; in Compute() local
|