Searched refs:input_batch_dimensions (Results 1 – 8 of 8) sorted by relevance
107 const std::vector<int>& input_batch_dimensions, in ApplyDelegate() argument110 TfLiteIntArrayCreate(input_batch_dimensions.size()); in ApplyDelegate()113 for (int i = 0; i < input_batch_dimensions.size(); ++i) in ApplyDelegate()114 input_batch_dim->data[i] = input_batch_dimensions[i]; in ApplyDelegate()120 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,
166 for (int i = 0; i < params_.input_batch_dimensions->size; ++i) { in ResizeOutputTensors()168 if (params_.input_batch_dimensions->data[i] == -1) continue; in ResizeOutputTensors()172 input_tensor->dims->data[params_.input_batch_dimensions->data[i]]; in ResizeOutputTensors()250 params_.input_batch_dimensions, in BuildGraph()
80 TfLiteIntArray* input_batch_dimensions; member