Home
last modified time | relevance | path

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

12

/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.py264 def segment_sum(data, segment_ids, num_segments, name=None): function
308 total = segment_sum(data, segment_ids, num_segments)
313 count = segment_sum(ones, segment_ids, num_segments)
324 total = segment_sum(data, segment_ids, num_segments)
329 count = segment_sum(ones, segment_ids, num_segments)
342 _set_ragged_segment_docstring(segment_sum, 'sum', 'summed')
Dragged_dispatch.py505 (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/lite/kernels/
Dsegment_sum.cc27 namespace segment_sum { namespace
129 static TfLiteRegistration r = {nullptr, nullptr, segment_sum::Prepare, in Register_SEGMENT_SUM()
130 segment_sum::Eval}; in Register_SEGMENT_SUM()
DAndroid.bp142 "segment_sum.cc",
DBUILD617 "segment_sum.cc",
/external/tensorflow/tensorflow/python/kernel_tests/
Dsegment_reduction_ops_test.py97 ops_list = [(np.add, None, math_ops.segment_sum),
105 complex_ops_list = [(np.add, None, math_ops.segment_sum),
138 math_ops.segment_sum(data=tf_x, segment_ids=indices)
147 s = math_ops.segment_sum(data=tf_x, segment_ids=indices)
159 result = math_ops.segment_sum(data=tf_x, segment_ids=indices).eval()
169 s = math_ops.segment_sum(data=tf_x, segment_ids=indices)
180 s = math_ops.segment_sum(data=tf_x, segment_ids=indices)
190 s = math_ops.segment_sum(data=tf_x, segment_ids=indices)
202 s = math_ops.segment_sum(data=tf_x, segment_ids=indices)
212 s = math_ops.segment_sum(data=tf_x, segment_ids=indices)
[all …]
/external/tensorflow/tensorflow/python/ops/
Dembedding_ops.py544 embeddings = math_ops.segment_sum(embeddings, segment_ids, name=name)
546 embeddings = math_ops.segment_sum(embeddings, segment_ids)
547 weight_sum = math_ops.segment_sum(weights, segment_ids)
550 embeddings = math_ops.segment_sum(embeddings, segment_ids)
552 weight_sum = math_ops.segment_sum(weights_squared, segment_ids)
Dmath_grad.py336 scaled_grad = math_ops.divide(grad, math_ops.segment_sum(ones, op.inputs[1]))
396 num_selected = math_ops.segment_sum(
437 num_zeros = gen_math_ops.segment_sum(
Dcontrol_flow_ops_test.py271 dynamic_grads = math_ops.segment_sum(dynamic_grads.values,
279 static_grads = math_ops.segment_sum(static_grads.values,
/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/security/advisory/
Dtfsa-2020-003.md12 …flow/blob/0e68f4d3295eb0281a517c3662f6698992b7b2cf/tensorflow/lite/kernels/segment_sum.cc#L39-L48):
Dtfsa-2020-002.md11 …flow/blob/0e68f4d3295eb0281a517c3662f6698992b7b2cf/tensorflow/lite/kernels/segment_sum.cc#L39-L44):
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.sparse.pbtxt92 name: "segment_sum"
Dtensorflow.math.pbtxt412 name: "segment_sum"
/external/tensorflow/tensorflow/python/distribute/
Dtpu_strategy_test.py710 segment_sum = math_ops.unsorted_segment_sum(
712 return sparse, segment_sum
760 segment_sum = math_ops.unsorted_segment_sum(
762 return {"sparse": sparse, "segment_sum": segment_sum}
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.sparse.pbtxt112 name: "segment_sum"
Dtensorflow.math.pbtxt412 name: "segment_sum"
Dtensorflow.pbtxt2064 name: "segment_sum"
/external/tensorflow/tensorflow/lite/
Dtflite_static.bp125 "kernels/segment_sum.cc",
/external/tensorflow/tensorflow/compiler/mlir/lite/tests/
Dlegalize-tf.mlir1975 // CHECK: "tfl.segment_sum"(%arg0, %arg1) : (tensor<3x3xf32>, tensor<i32>) -> tensor<*xf32>
1983 // CHECK: "tfl.segment_sum"

12