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

  /external/tensorflow/tensorflow/core/kernels/
quantized_mul_op.cc 192 int32 tensor_offset, int64 tensor_num_elements,
194 for (int i = 0; i < tensor_num_elements; ++i) {
205 const quint8* tensor_data, int32 tensor_offset, int64 tensor_num_elements,
209 CHECK_EQ(0, tensor_num_elements % vector_num_elements);
210 for (int base_i = 0; base_i < tensor_num_elements;
334 int64 tensor_num_elements; variable
341 tensor_num_elements = y.NumElements();
348 tensor_num_elements = x.NumElements();
353 tensor_offset, tensor_num_elements, z_data);
quantized_add_op.cc 293 float max_tensor, int64 tensor_num_elements,
295 for (int i = 0; i < tensor_num_elements; ++i) {
311 float max_tensor, int64 tensor_num_elements,
344 for (int64 base_i = 0; base_i < tensor_num_elements;
396 float max_tensor, int64 tensor_num_elements,
423 for (int i = 0; i < tensor_num_elements; ++i) {
520 int64 tensor_num_elements; variable
529 tensor_num_elements = y.NumElements();
538 tensor_num_elements = x.NumElements();
544 tensor_min, tensor_max, tensor_num_elements, min_z_value, max_z_value
    [all...]

Completed in 74 milliseconds