Lines Matching defs:memory_size
79 const uint32_t memory_size = SizeOfDimension(weights_time, 1);
92 stateShape->dimensions = { batch_size, memory_size * num_filters };
111 const int memory_size = SizeOfDimension(weights_time_, 1);
114 sizeof(float) * batch_size * memory_size * num_filters);
117 float* state_ptr_batch = GetBuffer<float>(state_out_) + b * memory_size * num_filters;
119 float* state_ptr = state_ptr_batch + c * memory_size;
120 state_ptr[memory_size - 1] = 0.0;
124 // is achieved by starting at state->data.f[memory_size - 1] and having the
125 // stride equal to memory_size.
129 &GetBuffer<float>(state_out_)[memory_size - 1], memory_size);
134 // stride equal to memory_size.
138 GetBuffer<float>(state_out_) + b * memory_size * num_filters;
141 GetBuffer<float>(weights_time_), state_out_ptr_batch, memory_size, num_filters,
174 GetBuffer<float>(state_out_) + b * memory_size * num_filters;
176 tflite::tensor_utils::VectorShiftLeft(state_out_ptr_batch, memory_size,
178 state_out_ptr_batch += memory_size;
