Searched refs:shrink_axis_mask (Results 1 – 8 of 8) sorted by relevance
| /AliOS-Things-master/components/ai_agent/src/engine/tflite-micro/tensorflow/lite/kernels/internal/ |
| A D | strided_slice_logic.h | 59 p->shrink_axis_mask <<= pad_count; in StridedSlicePadIndices()
126 const auto shrink_axis_mask = params.shrink_axis_mask; in StopForAxis() local
135 const bool shrink_axis = shrink_axis_mask & (1 << axis); in StopForAxis()
183 int begin_mask, int end_mask, int shrink_axis_mask, in BuildStridedSliceParams() argument
202 op_params.shrink_axis_mask = shrink_axis_mask; in BuildStridedSliceParams()
|
| A D | types.h | 1134 int16_t shrink_axis_mask; member
|
| /AliOS-Things-master/components/ai_agent/src/engine/tflite-micro/tensorflow/lite/micro/kernels/ |
| A D | strided_slice.cc | 78 op_params.shrink_axis_mask = op_context->params->shrink_axis_mask; in BuildStridedSliceParams()
103 const bool shrink_axis = op_context->params->shrink_axis_mask & (1 << idx); in CheckOutputSize()
|
| /AliOS-Things-master/components/ai_agent/src/engine/tflite-micro/tensorflow/lite/c/ |
| A D | builtin_op_data.h | 399 int shrink_axis_mask; member
|
| /AliOS-Things-master/components/ai_agent/src/engine/tflite-micro/tensorflow/lite/kernels/internal/reference/ |
| A D | legacy_reference_ops.h | 931 p->shrink_axis_mask = in StridedSliceReverseIndices()
932 LegacyReverseBits32(static_cast<uint32>(p->shrink_axis_mask)) >> in StridedSliceReverseIndices()
938 int begin_mask, int end_mask, int shrink_axis_mask, in StridedSlice() argument
945 begin_mask, end_mask, shrink_axis_mask, start_indices, stop_indices, in StridedSlice()
|
| /AliOS-Things-master/components/ai_agent/src/engine/tflite-micro/tensorflow/lite/core/api/ |
| A D | flatbuffer_conversions.cc | 1991 params->shrink_axis_mask = schema_params->shrink_axis_mask(); in ParseStridedSlice()
|
| /AliOS-Things-master/components/ai_agent/src/engine/tflite-micro/tensorflow/lite/schema/ |
| A D | schema_generated.h | 6911 int32_t shrink_axis_mask;
6917 shrink_axis_mask(0) {
6942 int32_t shrink_axis_mask() const {
6974 void add_shrink_axis_mask(int32_t shrink_axis_mask) {
6975 fbb_.AddElement<int32_t>(StridedSliceOptions::VT_SHRINK_AXIS_MASK, shrink_axis_mask, 0);
6995 int32_t shrink_axis_mask = 0) {
6997 builder_.add_shrink_axis_mask(shrink_axis_mask);
13355 { auto _e = shrink_axis_mask(); _o->shrink_axis_mask = _e; }
13370 auto _shrink_axis_mask = _o->shrink_axis_mask;
|
| /AliOS-Things-master/components/ai_agent/src/engine/tflite-micro/tensorflow/lite/kernels/internal/optimized/ |
| A D | legacy_optimized_ops.h | 4952 int begin_mask, int end_mask, int shrink_axis_mask, in StridedSlice() argument
4959 begin_mask, end_mask, shrink_axis_mask, start_indices, stop_indices, in StridedSlice()
|
Completed in 72 milliseconds