| /external/tensorflow/tensorflow/python/ops/ragged/ |
| D | ragged_segment_op_test.py | 51 def expected_value(self, data, segment_ids, num_segments, combiner): argument
101 def testRaggedSegment_Int(self, segment_op, combiner, segment_ids): argument
137 def testRaggedSegment_Float(self, segment_op, combiner, segment_ids): argument
|
| D | segment_id_ops.py | 76 def segment_ids_to_row_splits(segment_ids, num_segments=None, argument
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | unsorted_segment_test.h | 35 UnsortedSegmentModel(const TensorData& data, const TensorData& segment_ids, in UnsortedSegmentModel()
65 int segment_ids() const { return segment_ids_id_; } in segment_ids() function
80 const TensorData& segment_ids, in getModel()
|
| D | segment_sum.cc | 35 const TfLiteTensor* segment_ids, in ResizeOutputTensor()
70 const TfLiteTensor* segment_ids; in Prepare() local
92 const TfLiteTensor* segment_ids; in Eval() local
|
| D | unsorted_segment.cc | 45 const TfLiteTensor* segment_ids, in ResizeOutputTensor()
94 const TfLiteTensor* segment_ids; in Prepare() local
182 const TfLiteTensor* segment_ids; in EvalGeneric() local
|
| D | segment_sum_test.cc | 31 SegmentSumOpModel(const TensorData& data, const TensorData& segment_ids) { in SegmentSumOpModel()
40 int segment_ids() const { return segment_ids_id_; } in segment_ids() function in tflite::__anon07034af40111::SegmentSumOpModel
|
| D | unsorted_segment_prod_test.cc | 33 const TensorData& segment_ids, in UnsortedSegmentProdModel()
|
| D | unsorted_segment_min_test.cc | 34 UnsortedSegmentMinModel(const TensorData& data, const TensorData& segment_ids, in UnsortedSegmentMinModel()
|
| D | unsorted_segment_max_test.cc | 34 UnsortedSegmentMaxModel(const TensorData& data, const TensorData& segment_ids, in UnsortedSegmentMaxModel()
|
| D | unsorted_segment_sum_test.cc | 34 UnsortedSegmentSumModel(const TensorData& data, const TensorData& segment_ids, in UnsortedSegmentSumModel()
|
| /external/tflite-support/tensorflow_lite_support/custom_ops/kernel/ |
| D | unsorted_segment_test.h | 35 UnsortedSegmentModel(const TensorData& data, const TensorData& segment_ids, in UnsortedSegmentModel()
65 int segment_ids() const { return segment_ids_id_; } in segment_ids() function
80 const TensorData& segment_ids, in getModel()
|
| D | unsorted_segment.cc | 77 const TfLiteTensor* segment_ids, in ResizeOutputTensor()
126 const TfLiteTensor* segment_ids; in Prepare() local
214 const TfLiteTensor* segment_ids; in EvalGeneric() local
|
| D | unsorted_segment_prod_test.cc | 33 const TensorData& segment_ids, in UnsortedSegmentProdModel()
|
| D | unsorted_segment_max_test.cc | 34 UnsortedSegmentMaxModel(const TensorData& data, const TensorData& segment_ids, in UnsortedSegmentMaxModel()
|
| D | unsorted_segment_min_test.cc | 34 UnsortedSegmentMinModel(const TensorData& data, const TensorData& segment_ids, in UnsortedSegmentMinModel()
|
| D | unsorted_segment_sum_test.cc | 34 UnsortedSegmentSumModel(const TensorData& data, const TensorData& segment_ids, in UnsortedSegmentSumModel()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | segment_reduction_ops_impl_1.cc | 23 const Tensor& segment_ids) { in ValidateSegmentReduction()
44 const Tensor& segment_ids, in ValidateUnsortedSegmentReduction()
64 const Tensor& segment_ids, in ValidateSparseSegmentReduction()
|
| D | segment_reduction_ops_gpu.cu.h | 176 const int64_t output_outer_dim_size, const Index* __restrict__ segment_ids, in UnsortedSegmentCustomKernel()
196 const Tsegmentids* __restrict__ segment_ids, // [size] in SegmentOffsetsKernel()
226 const Tsegmentids* segment_ids, // [size] in LaunchSegmentOffsetsKernel()
|
| D | segment_reduction_ops_impl.h | 92 const Tensor& segment_ids = context->input(1); in Compute() local
232 const Tensor& segment_ids = context->input(1); in ComputeAsync() local
|
| /external/tensorflow/tensorflow/dtensor/mlir/expansions/ |
| D | segmentation_spmd_expander.cc | 96 mlir::Value segment_ids = sum_op.segment_ids(); in ExpandOp() 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
4681 segment_ids, argument
4757 segment_ids, argument
4826 segment_ids, argument
4872 segment_ids, argument
4911 segment_ids, argument
4948 segment_ids, argument
|
| D | string_ops.py | 586 segment_ids, argument
|
| /external/tensorflow/tensorflow/python/kernel_tests/strings_ops/ |
| D | unsorted_segment_join_op_test.py | 186 def test_multiple_cases_with_different_dims(self, inputs, segment_ids, argument
|
| /external/libaom/av1/common/ |
| D | seg_common.h | 99 static AOM_INLINE void set_segment_id(uint8_t *segment_ids, int mi_offset, in set_segment_id()
|
| /external/libvpx/vp9/common/ |
| D | vp9_pred_common.h | 23 const uint8_t *segment_ids, BLOCK_SIZE bsize, in get_segment_id()
|