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

  /external/tensorflow/tensorflow/contrib/lite/toco/graph_transformations/
propagate_fixed_sizes.cc 73 int rank_out = std::max(rank_x, rank_y); local
76 dims_out->reserve(rank_out);
77 for (int i = 0; i < rank_out; ++i) {
78 int dim_x = i < (rank_out - rank_x)
80 : input_shape_x.dims(i - (rank_out - rank_x));
81 bool dim_y_is_one = i < (rank_out - rank_y);
82 int dim_y = dim_y_is_one ? 1 : input_shape_y.dims(i - (rank_out - rank_y));
    [all...]
  /external/tensorflow/tensorflow/core/framework/
common_shape_fns.cc 1222 const int32 rank_out = std::max(rank_x, rank_y); local
    [all...]

Completed in 63 milliseconds