Searched refs:SegmentSum (Results 1 – 23 of 23) sorted by relevance
| /external/tensorflow/tensorflow/core/api_def/java_api/ |
| D | api_def_SegmentSum.pbtxt | 2 graph_op_name: "SegmentSum"
4 name: "math.SegmentSum"
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/ |
| D | SegmentSum.pbtxt | 2 name: "SegmentSum"
49 name: "SegmentSum"
98 name: "SegmentSum"
148 name: "SegmentSum"
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/ |
| D | SegmentSum.pbtxt | 2 name: "SegmentSum"
49 name: "SegmentSum"
98 name: "SegmentSum"
148 name: "SegmentSum"
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_SegmentSum.pbtxt | 2 graph_op_name: "SegmentSum"
30 <img style="width:100%" src="https://www.tensorflow.org/images/SegmentSum.png" alt>
|
| D | api_def_SparseSegmentSum.pbtxt | 28 Like `SegmentSum`, but `segment_ids` can have rank less than `data`'s first
|
| D | api_def_UnsortedSegmentSum.pbtxt | 25 that `segment_ids[j...] == i`. Unlike `SegmentSum`, `segment_ids`
|
| /external/tensorflow/tensorflow/core/api_def/python_api/ |
| D | api_def_SegmentSum.pbtxt | 2 graph_op_name: "SegmentSum"
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | segment_sum.cc | 105 reference_ops::SegmentSum<dtype>( \ in Eval()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | segment_reduction_ops_test.cc | 105 BM_Reduce(SegmentSum, R, C, S); \
|
| /external/tensorflow/tensorflow/cc/gradients/ |
| D | math_grad_test.cc | 48 using ops::SegmentSum;
957 TEST_F(NaryGradTest, SegmentSum) { in TEST_F() argument
960 auto y = SegmentSum(scope_, x, {0, 0, 1}); in TEST_F()
|
| /external/tensorflow/tensorflow/lite/g3doc/guide/ |
| D | op_select_allowlist.md | 469 * `raw_ops.SegmentSum`
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | reference_ops.h | 1799 inline void SegmentSum(const RuntimeShape& input_shape, const T* input_data, in SegmentSum() function
|
| /external/tensorflow/tensorflow/lite/toco/ |
| D | tooling_util.cc | 392 HANDLE_OPERATORTYPENAME_CASE(SegmentSum) in OperatorTypeName()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/tests/ |
| D | legalize-tf.mlir | 1972 %0 = "tf.SegmentSum"(%arg0, %arg1) : (tensor<3x3xf32>, tensor<i32>) -> tensor<*xf32>
1979 %0 = "tf.SegmentSum"(%arg0, %arg1) : (tensor<3x3xf32>, tensor<i64>) -> tensor<*xf32>
|
| /external/tensorflow/tensorflow/compiler/mlir/tosa/g3doc/ |
| D | legalization.md | 2633 ### tf.SegmentSum
|
| /external/tensorflow/tensorflow/core/ops/compat/ |
| D | ops_history.v0.pbtxt | 30212 name: "SegmentSum"
30252 name: "SegmentSum"
30293 name: "SegmentSum"
30335 name: "SegmentSum"
|
| D | ops_history.v2.pbtxt | 67649 name: "SegmentSum"
67696 name: "SegmentSum"
67745 name: "SegmentSum"
67795 name: "SegmentSum"
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/ir/ |
| D | tfl_ops.td | 4373 let summary = "SegmentSum operator";
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_generated_ops.td | 13138 def TF_SegmentSumOp : TF_Op<"SegmentSum", [NoSideEffect]> {
13153 <img style="width:100%" src="https://www.tensorflow.org/images/SegmentSum.png" alt>
14409 Like `SegmentSum`, but `segment_ids` can have rank less than `data`'s first
17343 that `segment_ids[j...] == i`. Unlike `SegmentSum`, `segment_ids`
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.raw_ops.pbtxt | 4004 name: "SegmentSum"
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.raw_ops.pbtxt | 4004 name: "SegmentSum"
|
| /external/tensorflow/tensorflow/go/op/ |
| D | wrappers.go | 22512 func SegmentSum(scope *Scope, data tf.Output, segment_ids tf.Output) (output tf.Output) { func
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | ops.pbtxt | 45282 name: "SegmentSum"
|