| /external/tensorflow/tensorflow/compiler/tests/ |
| D | segment_reduction_ops_test.py | 31 def _segmentReduction(self, op, data, indices, num_segments): argument
40 def _unsortedSegmentSum(self, data, indices, num_segments): argument
44 def _unsortedSegmentProd(self, data, indices, num_segments): argument
48 def _unsortedSegmentMin(self, data, indices, num_segments): argument
52 def _unsortedSegmentMax(self, data, indices, num_segments): argument
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | unsorted_segment_test.h | 36 const TensorData& num_segments, const BuiltinOperator op, in UnsortedSegmentModel()
66 int num_segments() const { return num_segments_id_; } in num_segments() function
81 const TensorData& num_segments) { in getModel()
|
| D | unsorted_segment.cc | 46 const TfLiteTensor* num_segments, in ResizeOutputTensor()
97 const TfLiteTensor* num_segments; in Prepare() local
185 const TfLiteTensor* num_segments; in EvalGeneric() local
|
| D | unsorted_segment_prod_test.cc | 34 const TensorData& num_segments) in UnsortedSegmentProdModel()
|
| D | unsorted_segment_min_test.cc | 35 const TensorData& num_segments) in UnsortedSegmentMinModel()
|
| D | unsorted_segment_max_test.cc | 35 const TensorData& num_segments) in UnsortedSegmentMaxModel()
|
| D | unsorted_segment_sum_test.cc | 35 const TensorData& num_segments) in UnsortedSegmentSumModel()
|
| /external/tflite-support/tensorflow_lite_support/custom_ops/kernel/ |
| D | unsorted_segment_test.h | 36 const TensorData& num_segments, const BuiltinOperator op, in UnsortedSegmentModel()
66 int num_segments() const { return num_segments_id_; } in num_segments() function
81 const TensorData& num_segments) { in getModel()
|
| D | unsorted_segment.cc | 78 const TfLiteTensor* num_segments, in ResizeOutputTensor()
129 const TfLiteTensor* num_segments; in Prepare() local
217 const TfLiteTensor* num_segments; in EvalGeneric() local
|
| D | unsorted_segment_prod_test.cc | 34 const TensorData& num_segments) in UnsortedSegmentProdModel()
|
| D | unsorted_segment_max_test.cc | 35 const TensorData& num_segments) in UnsortedSegmentMaxModel()
|
| D | unsorted_segment_min_test.cc | 35 const TensorData& num_segments) in UnsortedSegmentMinModel()
|
| D | unsorted_segment_sum_test.cc | 35 const TensorData& num_segments) in UnsortedSegmentSumModel()
|
| /external/tensorflow/tensorflow/python/ops/ragged/ |
| D | ragged_math_ops.py | 164 num_segments, argument
269 num_segments, argument
283 num_segments, argument
297 num_segments, argument
311 num_segments, argument
325 num_segments, argument
345 num_segments, argument
|
| D | segment_id_ops.py | 76 def segment_ids_to_row_splits(segment_ids, num_segments=None, argument
|
| D | ragged_segment_op_test.py | 51 def expected_value(self, data, segment_ids, num_segments, combiner): argument
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | unsorted_segment_join_op.cc | 62 const int64_t num_segments) { in GetOutputShape()
98 auto num_segments = num_segments_tensor.scalar<NUM_SEGMENTS_TYPE>()(); in Compute() local
|
| D | segment_reduction_ops_impl_1.cc | 45 const Tensor& num_segments) { in ValidateUnsortedSegmentReduction()
|
| D | gpu_prim_helpers.h | 213 OpKernelContext* context, int num_segments, ReduceOp reduce_op, in GpuSegmentedReduce()
|
| D | segment_reduction_ops_test.cc | 60 Tensor num_segments(DT_INT32, TensorShape({})); in BM_UnsortedSegmentReduction() local
|
| /external/tensorflow/tensorflow/core/tpu/kernels/xla/ |
| D | segment_reduction_ops.cc | 50 int64_t num_segments; in Compile() local
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | math_ops.py | 4542 def _unsorted_segment_N(data, segment_ids, num_segments): argument
4569 def unsorted_segment_mean(data, segment_ids, num_segments, name=None): argument
4624 def unsorted_segment_sqrt_n(data, segment_ids, num_segments, name=None): argument
4683 num_segments=None): argument
4758 num_segments=None, argument
4828 num_segments=None): argument
4873 num_segments=None, argument
4913 num_segments=None): argument
4949 num_segments=None, argument
|
| /external/tensorflow/tensorflow/python/kernel_tests/strings_ops/ |
| D | unsorted_segment_join_op_test.py | 187 num_segments, separator, argument
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | segment_reduction_ops.cc | 60 int64_t num_segments; in Compile() local
|
| /external/tensorflow/tensorflow/python/kernel_tests/math_ops/ |
| D | segment_reduction_ops_test.py | 48 def _segmentReduce(self, indices, x, op1, op2=None, num_segments=None, argument
546 num_segments=None): argument
|