Searched refs:unsorted_segment_mean (Results 1 – 11 of 11) sorted by relevance
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | segment_reduction_ops_deterministic_test.py | 86 math_ops.unsorted_segment_mean: True, # uses unsorted_segment_sum
|
| /external/tensorflow/tensorflow/python/kernel_tests/math_ops/ |
| D | segment_reduction_ops_d9m_test.py | 95 math_ops.unsorted_segment_mean: True, # uses unsorted_segment_sum
|
| D | segment_reduction_ops_test.py | 291 math_ops.unsorted_segment_mean, lambda t: 0),
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.math.pbtxt | 492 name: "unsorted_segment_mean"
|
| D | tensorflow.pbtxt | 2484 name: "unsorted_segment_mean"
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.math.pbtxt | 492 name: "unsorted_segment_mean"
|
| /external/tensorflow/tensorflow/python/ops/ragged/ |
| D | ragged_math_ops.py | 322 @dispatch.dispatch_for_api(math_ops.unsorted_segment_mean)
|
| D | ragged_dispatch_test.py | 838 op=math_ops.unsorted_segment_mean,
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | nn_grad.py | 1218 means_i = math_ops.unsorted_segment_mean(
|
| D | math_ops.py | 4569 def unsorted_segment_mean(data, segment_ids, num_segments, name=None): function
|
| /external/tensorflow/ |
| D | RELEASE.md | 605 * (since v2.7) `tf.math.unsorted_segment_mean`
1448 * `tf.math.unsorted_segment_mean`
3924 `tf.math.segment_sum`, `tf.math.unsorted_segment_mean`,
|