Home
last modified time | relevance | path

Searched refs:segment_sum (Results 1 – 21 of 21) sorted by relevance

/external/tensorflow/tensorflow/python/ops/ragged/
Dragged_segment_op_test.py84 (ragged_math_ops.segment_sum, sum, [0, 0, 1, 1, 2, 2]),
85 (ragged_math_ops.segment_sum, sum, [0, 0, 0, 1, 1, 1]),
86 (ragged_math_ops.segment_sum, sum, [5, 4, 3, 2, 1, 0]),
87 (ragged_math_ops.segment_sum, sum, [0, 0, 0, 10, 10, 10]),
116 (ragged_math_ops.segment_sum, sum, [0, 0, 1, 1, 2, 2]),
117 (ragged_math_ops.segment_sum, sum, [0, 0, 0, 1, 1, 1]),
118 (ragged_math_ops.segment_sum, sum, [5, 4, 3, 2, 1, 0]),
119 (ragged_math_ops.segment_sum, sum, [0, 0, 0, 10, 10, 10]),
159 segmented1 = ragged_math_ops.segment_sum(rt, segment_ids1, 3)
167 segmented2 = ragged_math_ops.segment_sum(rt, segment_ids2, 3)
[all …]
Dragged_math_ops.py245 def segment_sum(data, segment_ids, num_segments, name=None): function
277 total = segment_sum(data, segment_ids, num_segments)
280 count = segment_sum(ones, segment_ids, num_segments)
291 total = segment_sum(data, segment_ids, num_segments)
294 count = segment_sum(ones, segment_ids, num_segments)
307 _set_ragged_segment_docstring(segment_sum, 'sum', 'summed')
Dragged_dispatch.py443 (math_ops.unsorted_segment_sum, ragged_math_ops.segment_sum,
/external/tensorflow/tensorflow/core/api_def/python_api/
Dapi_def_SegmentSum.pbtxt4 name: "math.segment_sum"
7 name: "segment_sum"
/external/tensorflow/tensorflow/python/kernel_tests/
Dsegment_reduction_ops_test.py95 ops_list = [(np.add, None, math_ops.segment_sum),
103 complex_ops_list = [(np.add, None, math_ops.segment_sum),
136 math_ops.segment_sum(data=tf_x, segment_ids=indices)
145 s = math_ops.segment_sum(data=tf_x, segment_ids=indices)
157 result = math_ops.segment_sum(data=tf_x, segment_ids=indices).eval()
167 s = math_ops.segment_sum(data=tf_x, segment_ids=indices)
178 s = math_ops.segment_sum(data=tf_x, segment_ids=indices)
188 s = math_ops.segment_sum(data=tf_x, segment_ids=indices)
200 s = math_ops.segment_sum(data=tf_x, segment_ids=indices)
210 s = math_ops.segment_sum(data=tf_x, segment_ids=indices)
[all …]
/external/tensorflow/tensorflow/python/ops/
Dembedding_ops.py515 embeddings = math_ops.segment_sum(embeddings, segment_ids, name=name)
517 embeddings = math_ops.segment_sum(embeddings, segment_ids)
518 weight_sum = math_ops.segment_sum(weights, segment_ids)
521 embeddings = math_ops.segment_sum(embeddings, segment_ids)
523 weight_sum = math_ops.segment_sum(weights_squared, segment_ids)
Dcontrol_flow_ops_test.py257 dynamic_grads = math_ops.segment_sum(dynamic_grads.values,
265 static_grads = math_ops.segment_sum(static_grads.values,
Dmath_grad.py203 scaled_grad = math_ops.divide(grad, math_ops.segment_sum(ones, op.inputs[1]))
263 num_selected = math_ops.segment_sum(math_ops.cast(is_selected, grad.dtype),
/external/tensorflow/tensorflow/core/api_def/base_api/
Dapi_def_SparseSegmentSqrtN.pbtxt26 See `tf.sparse.segment_sum` for usage examples.
Dapi_def_SparseSegmentMean.pbtxt24 See `tf.sparse.segment_sum` for usage examples.
Dapi_def_SegmentSum.pbtxt37 tf.segment_sum(c, tf.constant([0, 0, 1]))
Dapi_def_SparseSegmentSum.pbtxt51 tf.segment_sum(c, tf.constant([0, 0, 1]))
/external/tensorflow/tensorflow/contrib/layers/python/layers/
Dembedding_ops.py696 weight_sum = math_ops.segment_sum(weights, segment_ids)
700 weight_sum = math_ops.segment_sum(weights_squared, segment_ids)
767 return math_ops.segment_sum(ret, segment_ids, name=name)
898 partitioned_result[p] = math_ops.segment_sum(
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.sparse.pbtxt84 name: "segment_sum"
Dtensorflow.math.pbtxt380 name: "segment_sum"
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.sparse.pbtxt104 name: "segment_sum"
Dtensorflow.math.pbtxt380 name: "segment_sum"
Dtensorflow.pbtxt1980 name: "segment_sum"
/external/tensorflow/tensorflow/contrib/linear_optimizer/python/ops/
Dsdca_ops.py314 result_sparse += math_ops.segment_sum(
/external/tensorflow/tensorflow/contrib/metrics/python/ops/
Dmetric_ops.py1304 weight_totals_for_true = math_ops.segment_sum(weights_for_false,
1306 weight_totals_for_false = math_ops.segment_sum(weights_for_true,
/external/tensorflow/
DRELEASE.md359 …segment_mean`, `tf.math.segment_min`, `tf.math.segment_prod`, `tf.math.segment_sum`, `tf.math.sin`…