Home
last modified time | relevance | path

Searched refs:batch_dims (Results 1 – 6 of 6) sorted by relevance

/AliOS-Things-master/components/ai_agent/src/engine/tflite-micro/tensorflow/lite/kernels/internal/reference/
A Dgather.h38 int batch_dims = op_params.batch_dims; in Gather() local
39 if (batch_dims < 0) { in Gather()
40 batch_dims += coords_shape.DimensionsCount(); in Gather()
42 TFLITE_DCHECK_GE(batch_dims, 0); in Gather()
43 TFLITE_DCHECK_LT(batch_dims, input_shape.DimensionsCount()); in Gather()
44 TFLITE_DCHECK_LE(batch_dims, coords_shape.DimensionsCount()); in Gather()
45 TFLITE_DCHECK_GE(axis, batch_dims); in Gather()
46 for (int i = 0; i < batch_dims; ++i) { in Gather()
53 for (int i = 0; i < batch_dims; ++i) { in Gather()
58 for (int i = batch_dims; i < axis; ++i) { in Gather()
[all …]
A Dlegacy_reference_ops.h898 op_params.batch_dims = 0; in Gather()
/AliOS-Things-master/components/ai_agent/src/engine/tflite-micro/tensorflow/lite/c/
A Dbuiltin_op_data.h368 int batch_dims; member
/AliOS-Things-master/components/ai_agent/src/engine/tflite-micro/tensorflow/lite/core/api/
A Dflatbuffer_conversions.cc1336 params->batch_dims = 0; in ParseGather()
1339 params->batch_dims = gather_params->batch_dims(); in ParseGather()
/AliOS-Things-master/components/ai_agent/src/engine/tflite-micro/tensorflow/lite/kernels/internal/
A Dtypes.h955 int16_t batch_dims; member
/AliOS-Things-master/components/ai_agent/src/engine/tflite-micro/tensorflow/lite/schema/
A Dschema_generated.h6497 int32_t batch_dims;
6500 batch_dims(0) {
6513 int32_t batch_dims() const {
6533 void add_batch_dims(int32_t batch_dims) {
6534 fbb_.AddElement<int32_t>(GatherOptions::VT_BATCH_DIMS, batch_dims, 0);
6551 int32_t batch_dims = 0) {
6553 builder_.add_batch_dims(batch_dims);
13150 { auto _e = batch_dims(); _o->batch_dims = _e; }
13162 auto _batch_dims = _o->batch_dims;

Completed in 42 milliseconds