Searched defs:SegmentSum (Results 1 – 3 of 3) sorted by relevance
957 TEST_F(NaryGradTest, SegmentSum) { in TEST_F() argument
1799 inline void SegmentSum(const RuntimeShape& input_shape, const T* input_data, in SegmentSum() function
22512 func SegmentSum(scope *Scope, data tf.Output, segment_ids tf.Output) (output tf.Output) { func