Searched defs:accum_depth (Results 1 - 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/contrib/lite/kernels/internal/reference/
H A Dreference_ops.h460 const int accum_depth = ArraySize(weights_dims, 0); local
466 for (int d = 0; d < accum_depth; ++d) {
467 total += input_data[b * accum_depth + d] *
468 weights_data[out_c * accum_depth + d];
512 const int accum_depth = ArraySize(filter_dims, 0); local
518 for (int d = 0; d < accum_depth; ++d) {
519 int32 input_val = input_data[b * accum_depth + d];
520 int32 filter_val = filter_data[out_c * accum_depth + d];

Completed in 80 milliseconds