Searched defs:group_dim (Results 1 – 3 of 3) sorted by relevance
77 int32_t group_dim; in add_reduce_node() local
81 int32_t group_dim; in add_softmax_node() local
449 const auto group_dim = group_shape[group_dim_index]; in PopulateGroupIndices() local