Searched refs:input_batch_dimensions (Results 1 – 8 of 8) sorted by relevance
106 const std::vector<int>& input_batch_dimensions, in ApplyDelegate() argument109 TfLiteIntArrayCreate(input_batch_dimensions.size()); in ApplyDelegate()112 for (int i = 0; i < input_batch_dimensions.size(); ++i) in ApplyDelegate()113 input_batch_dim->data[i] = input_batch_dimensions[i]; in ApplyDelegate()119 options.input_batch_dimensions = input_batch_dim; in ApplyDelegate()
43 void SetInputBatchDimensions(TfLiteIntArray* input_batch_dimensions) { in SetInputBatchDimensions() argument44 input_batch_dims_ = input_batch_dimensions; in SetInputBatchDimensions()
58 TfLiteIntArray* input_batch_dimensions, in CreateBatchSeqBuilder() argument62 builder->SetInputBatchDimensions(input_batch_dimensions); in CreateBatchSeqBuilder()
53 TfLiteIntArray* input_batch_dimensions,
408 TfLiteIntArray* input_batch_dimensions, in AddBatchSeqConfig() argument412 input_batch_dimensions, output_batch_dimensions); in AddBatchSeqConfig()
266 TfLiteIntArray* input_batch_dimensions,
164 for (int i = 0; i < params_.input_batch_dimensions->size; ++i) { in ResizeOutputTensors()166 if (params_.input_batch_dimensions->data[i] == -1) continue; in ResizeOutputTensors()170 input_tensor->dims->data[params_.input_batch_dimensions->data[i]]; in ResizeOutputTensors()246 params_.input_batch_dimensions, in BuildGraph()
80 TfLiteIntArray* input_batch_dimensions; member