Searched refs:arg_dim_in_output (Results 1 – 1 of 1) sorted by relevance
3022 std::vector<bool> arg_dim_in_output(arg->shape().rank(), true); in HandleReduce() local3025 arg_dim_in_output[dim] = false; in HandleReduce()3034 for (int64 i = 0; i < arg_dim_in_output.size(); ++i) { in HandleReduce()3035 if (arg_dim_in_output[i] && !arg_dim_unmodified[i]) { in HandleReduce()3043 if (arg_dim_in_output[dim_pair.second]) { in HandleReduce()