| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tests/ |
| H A 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
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/ops/ragged/ |
| H A 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
|
| H A D | segment_id_ops.py | 76 def segment_ids_to_row_splits(segment_ids, num_segments=None, argument
|
| H A D | ragged_segment_op_test.py | 51 def expected_value(self, data, segment_ids, num_segments, combiner): argument
|
| /aosp_15_r20/external/tflite-support/tensorflow_lite_support/custom_ops/kernel/ |
| H A 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()
|
| H A 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
|
| H A D | unsorted_segment_prod_test.cc | 34 const TensorData& num_segments) in UnsortedSegmentProdModel()
|
| H A D | unsorted_segment_min_test.cc | 35 const TensorData& num_segments) in UnsortedSegmentMinModel()
|
| H A D | unsorted_segment_max_test.cc | 35 const TensorData& num_segments) in UnsortedSegmentMaxModel()
|
| H A D | unsorted_segment_sum_test.cc | 35 const TensorData& num_segments) in UnsortedSegmentSumModel()
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/ |
| H A 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()
|
| H A 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
|
| H A D | unsorted_segment_prod_test.cc | 34 const TensorData& num_segments) in UnsortedSegmentProdModel()
|
| H A D | unsorted_segment_min_test.cc | 35 const TensorData& num_segments) in UnsortedSegmentMinModel()
|
| H A D | unsorted_segment_max_test.cc | 35 const TensorData& num_segments) in UnsortedSegmentMaxModel()
|
| H A D | unsorted_segment_sum_test.cc | 35 const TensorData& num_segments) in UnsortedSegmentSumModel()
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/ops/ |
| H A 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
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A 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
|
| H A D | segment_reduction_ops_impl_1.cc | 45 const Tensor& num_segments) { in ValidateUnsortedSegmentReduction()
|
| /aosp_15_r20/external/executorch/runtime/executor/ |
| H A D | program.cpp | 404 size_t num_segments = internal_program_->segments()->size(); in LoadSegment() local
469 size_t num_segments = internal_program_->segments()->size(); in load_mutable_subsegment_into() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/math_ops/ |
| H A D | segment_reduction_ops_test.py | 48 def _segmentReduce(self, indices, x, op1, op2=None, num_segments=None, argument
546 num_segments=None): argument
|
| /aosp_15_r20/external/webp/src/enc/ |
| H A D | quant_enc.c | 245 const int num_segments = enc->segment_hdr_.num_segments_; in SetupMatrices() local
367 const int num_segments = (enc->segment_hdr_.num_segments_ < NUM_MB_SEGMENTS) in SimplifySegments() local
405 const int num_segments = enc->segment_hdr_.num_segments_; in VP8SetSegmentParams() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/tpu/kernels/xla/ |
| H A D | segment_reduction_ops.cc | 50 int64_t num_segments; in Compile() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/strings_ops/ |
| H A D | unsorted_segment_join_op_test.py | 187 num_segments, separator, argument
|
| /aosp_15_r20/external/mesa3d/src/imgui/ |
| H A D | imgui_draw.cpp | 953 …DrawList::PathArcTo(const ImVec2& centre, float radius, float a_min, float a_max, int num_segments) in PathArcTo()
997 …DrawList::PathBezierCurveTo(const ImVec2& p2, const ImVec2& p3, const ImVec2& p4, int num_segments) in PathBezierCurveTo()
1144 void ImDrawList::AddCircle(const ImVec2& centre, float radius, ImU32 col, int num_segments, float t… in AddCircle()
1155 void ImDrawList::AddCircleFilled(const ImVec2& centre, float radius, ImU32 col, int num_segments) in AddCircleFilled()
1166 …t ImVec2& cp0, const ImVec2& cp1, const ImVec2& pos1, ImU32 col, float thickness, int num_segments) in AddBezierCurve()
|