Searched defs:segment_id_shape (Results 1 – 11 of 11) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | unsorted_segment_join_op.cc | 47 const TensorShape& input_shape, const TensorShape& segment_id_shape) { in GetStrides()
61 const TensorShape& segment_id_shape, in GetOutputShape()
89 const TensorShape& segment_id_shape = segment_id.shape(); in Compute() local
|
| /external/tflite-support/tensorflow_lite_support/custom_ops/kernel/ |
| D | unsorted_segment_test.h | 49 const std::initializer_list<int>& segment_id_shape, in UnsortedSegmentModel()
87 const std::initializer_list<int>& segment_id_shape, in getConstModel()
|
| D | unsorted_segment_prod_test.cc | 41 const std::initializer_list<int>& segment_id_shape, in UnsortedSegmentProdModel()
|
| D | unsorted_segment_min_test.cc | 42 const std::initializer_list<int>& segment_id_shape, in UnsortedSegmentMinModel()
|
| D | unsorted_segment_max_test.cc | 42 const std::initializer_list<int>& segment_id_shape, in UnsortedSegmentMaxModel()
|
| D | unsorted_segment_sum_test.cc | 42 const std::initializer_list<int>& segment_id_shape, in UnsortedSegmentSumModel()
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | unsorted_segment_test.h | 49 const std::initializer_list<int>& segment_id_shape, in UnsortedSegmentModel()
87 const std::initializer_list<int>& segment_id_shape, in getConstModel()
|
| D | unsorted_segment_prod_test.cc | 41 const std::initializer_list<int>& segment_id_shape, in UnsortedSegmentProdModel()
|
| D | unsorted_segment_min_test.cc | 42 const std::initializer_list<int>& segment_id_shape, in UnsortedSegmentMinModel()
|
| D | unsorted_segment_max_test.cc | 42 const std::initializer_list<int>& segment_id_shape, in UnsortedSegmentMaxModel()
|
| D | unsorted_segment_sum_test.cc | 42 const std::initializer_list<int>& segment_id_shape, in UnsortedSegmentSumModel()
|