Home
last modified time | relevance | path

Searched refs:batch_dim (Results 1 – 4 of 4) sorted by relevance

/AliOS-Things-master/components/ai_agent/src/engine/tflite-micro/tensorflow/lite/c/
A Dbuiltin_op_data.h462 int batch_dim; member
/AliOS-Things-master/components/ai_agent/src/engine/tflite-micro/tensorflow/lite/kernels/internal/reference/
A Dreference_ops.h1419 const int batch_dim, const RuntimeShape& input_shape, in ReverseSequence() argument
1425 int outer_dim = std::min(batch_dim, seq_dim); in ReverseSequence()
1426 int medium_dim = std::max(batch_dim, seq_dim); in ReverseSequence()
1445 if (batch_dim > seq_dim) { in ReverseSequence()
1469 } else if (batch_dim < seq_dim) { in ReverseSequence()
/AliOS-Things-master/components/ai_agent/src/engine/tflite-micro/tensorflow/lite/core/api/
A Dflatbuffer_conversions.cc728 params->batch_dim = reverse_seq_params->batch_dim(); in ParseOpDataTfLite()
/AliOS-Things-master/components/ai_agent/src/engine/tflite-micro/tensorflow/lite/schema/
A Dschema_generated.h9088 int32_t batch_dim;
9091 batch_dim(0) {
9104 int32_t batch_dim() const {
9124 void add_batch_dim(int32_t batch_dim) {
9125 fbb_.AddElement<int32_t>(ReverseSequenceOptions::VT_BATCH_DIM, batch_dim, 0);
9142 int32_t batch_dim = 0) {
9144 builder_.add_batch_dim(batch_dim);
14488 { auto _e = batch_dim(); _o->batch_dim = _e; }
14500 auto _batch_dim = _o->batch_dim;

Completed in 31 milliseconds