Home
last modified time | relevance | path

Searched refs:sparse_segment_sum (Results 1 – 6 of 6) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/math_ops/
H A Dsegment_reduction_ops_test.py581 ops_list = [(np.add, None, math_ops.sparse_segment_sum),
622 ops_list = [(np.add, None, math_ops.sparse_segment_sum), (
682 math_ops.sparse_segment_sum,
694 ops_list = [(np.add, None, math_ops.sparse_segment_sum), (
709 ops_list = [math_ops.sparse_segment_sum, math_ops.sparse_segment_mean]
720 ops_list = [math_ops.sparse_segment_sum, math_ops.sparse_segment_mean]
733 ops_list = [math_ops.sparse_segment_sum, math_ops.sparse_segment_mean]
746 ops_list = [math_ops.sparse_segment_sum, math_ops.sparse_segment_mean]
758 ops_list = [math_ops.sparse_segment_sum, math_ops.sparse_segment_mean]
772 ops_list = [math_ops.sparse_segment_sum, math_ops.sparse_segment_mean]
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/core/api_def/base_api/
H A Dapi_def_SparseSegmentSum.pbtxt37 tf.sparse_segment_sum(c, tf.constant([0, 1]), tf.constant([0, 0]))
41 tf.sparse_segment_sum(c, tf.constant([0, 1]), tf.constant([0, 1]))
46 tf.sparse_segment_sum(c, tf.constant([0, 1, 2]), tf.constant([0, 0, 1]))
/aosp_15_r20/external/tensorflow/tensorflow/python/ops/
H A Dembedding_ops.py577 embeddings = math_ops.sparse_segment_sum(
H A Dmath_ops.py4679 def sparse_segment_sum(data, function
4750 return gen_math_ops.sparse_segment_sum(
4818 return sparse_segment_sum(
/aosp_15_r20/external/tensorflow/tensorflow/tools/api/golden/v1/
H A Dtensorflow.pbtxt2232 name: "sparse_segment_sum"
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
H A Dtf_generated_ops.td16196 tf.sparse_segment_sum(c, tf.constant([0, 1]), tf.constant([0, 0]))
16200 tf.sparse_segment_sum(c, tf.constant([0, 1]), tf.constant([0, 1]))
16205 tf.sparse_segment_sum(c, tf.constant([0, 1, 2]), tf.constant([0, 0, 1]))