Home
last modified time | relevance | path

Searched defs:segment_ids (Results 1 – 25 of 35) sorted by relevance

12

/aosp_15_r20/external/tensorflow/tensorflow/python/ops/ragged/
H A Dragged_segment_op_test.py51 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
H A Dsegment_id_ops.py76 def segment_ids_to_row_splits(segment_ids, num_segments=None, argument
/aosp_15_r20/external/tflite-support/tensorflow_lite_support/custom_ops/kernel/
H A Dunsorted_segment_test.h35 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()
H A Dunsorted_segment.cc77 const TfLiteTensor* segment_ids, in ResizeOutputTensor()
126 const TfLiteTensor* segment_ids; in Prepare() local
214 const TfLiteTensor* segment_ids; in EvalGeneric() local
H A Dunsorted_segment_prod_test.cc33 const TensorData& segment_ids, in UnsortedSegmentProdModel()
H A Dunsorted_segment_min_test.cc34 UnsortedSegmentMinModel(const TensorData& data, const TensorData& segment_ids, in UnsortedSegmentMinModel()
H A Dunsorted_segment_max_test.cc34 UnsortedSegmentMaxModel(const TensorData& data, const TensorData& segment_ids, in UnsortedSegmentMaxModel()
H A Dunsorted_segment_sum_test.cc34 UnsortedSegmentSumModel(const TensorData& data, const TensorData& segment_ids, in UnsortedSegmentSumModel()
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/
H A Dunsorted_segment_test.h35 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()
H A Dsegment_sum.cc35 const TfLiteTensor* segment_ids, in ResizeOutputTensor()
70 const TfLiteTensor* segment_ids; in Prepare() local
92 const TfLiteTensor* segment_ids; in Eval() local
H A Dunsorted_segment.cc45 const TfLiteTensor* segment_ids, in ResizeOutputTensor()
94 const TfLiteTensor* segment_ids; in Prepare() local
182 const TfLiteTensor* segment_ids; in EvalGeneric() local
H A Dsegment_sum_test.cc31 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::__anond29409850111::SegmentSumOpModel
H A Dunsorted_segment_prod_test.cc33 const TensorData& segment_ids, in UnsortedSegmentProdModel()
H A Dunsorted_segment_min_test.cc34 UnsortedSegmentMinModel(const TensorData& data, const TensorData& segment_ids, in UnsortedSegmentMinModel()
H A Dunsorted_segment_max_test.cc34 UnsortedSegmentMaxModel(const TensorData& data, const TensorData& segment_ids, in UnsortedSegmentMaxModel()
H A Dunsorted_segment_sum_test.cc34 UnsortedSegmentSumModel(const TensorData& data, const TensorData& segment_ids, in UnsortedSegmentSumModel()
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dsegment_reduction_ops_impl_1.cc23 const Tensor& segment_ids) { in ValidateSegmentReduction()
44 const Tensor& segment_ids, in ValidateUnsortedSegmentReduction()
64 const Tensor& segment_ids, in ValidateSparseSegmentReduction()
H A Dsegment_reduction_ops_gpu.cu.h176 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()
H A Dsegment_reduction_ops_impl.h92 const Tensor& segment_ids = context->input(1); in Compute() local
232 const Tensor& segment_ids = context->input(1); in ComputeAsync() local
/aosp_15_r20/external/tensorflow/tensorflow/python/ops/
H A Dmath_ops.py4542 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
/aosp_15_r20/external/tensorflow/tensorflow/cc/gradients/
H A Dmath_grad_test.cc1067 auto segment_ids = Const(scope_, {0, 0, 1}); in TEST_F() local
1076 auto segment_ids = Const(scope_, {0ll, 0ll, 1ll}); in TEST_F() local
1085 auto segment_ids = Const(scope_, {0, 0, -1}); in TEST_F() local
1094 auto segment_ids = Const(scope_, {0, 0, 1}); in TEST_F() local
1103 auto segment_ids = Const(scope_, {0, 0, 1}); in TEST_F() local
/aosp_15_r20/external/tensorflow/tensorflow/dtensor/mlir/expansions/
H A Dsegmentation_spmd_expander.cc96 mlir::Value segment_ids = sum_op.segment_ids(); in ExpandOp() local
/aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/strings_ops/
H A Dunsorted_segment_join_op_test.py186 def test_multiple_cases_with_different_dims(self, inputs, segment_ids, argument
/aosp_15_r20/external/libaom/av1/common/
H A Dseg_common.h99 static inline void set_segment_id(uint8_t *segment_ids, int mi_offset, in set_segment_id()
/aosp_15_r20/external/libvpx/vp9/common/
H A Dvp9_pred_common.h23 const uint8_t *segment_ids, BLOCK_SIZE bsize, in get_segment_id()

12