HomeSort by relevance Sort by last modified time
    Searched full:num_elements_reduced (Results 1 - 3 of 3) sorted by null

  /external/tensorflow/tensorflow/compiler/tf2xla/kernels/
reduction_ops_common.cc 45 int64 num_elements_reduced) {
73 int64 num_elements_reduced = 1LL; local
84 num_elements_reduced *= data_shape.dim_size(index);
125 BuildFinalizer(ctx->builder(), reduce, num_elements_reduced);
reduction_ops.h 53 // output of the reduction. 'num_elements_reduced' is the number of elements
59 int64 num_elements_reduced);
reduction_ops.cc 112 int64 num_elements_reduced) override {
114 num_elements_reduced);

Completed in 153 milliseconds