Searched defs:outer_dim (Results 1 – 2 of 2) sorted by relevance
32 auto outer_dim = param.outer_dim_; in WorkerForReduceSparseGradient() local63 size_t first_dim, size_t outer_dim) { in ReduceSparseGradient()130 size_t outer_dim) { in ReduceMultiSparseGradient()158 size_t outer_dim) { in TwoLevelReduceSparseGradient()
965 …auto outer_dim = std::accumulate(indices_shape.begin(), indices_shape.end() - 1, 1, std::multiplie… in __anond7f8849a0b02() local