Searched refs:output_batch_dimensions (Results 1 – 8 of 8) sorted by relevance
107 const std::vector<int>& output_batch_dimensions) { in ApplyDelegate() argument111 TfLiteIntArrayCreate(output_batch_dimensions.size()); in ApplyDelegate()114 for (int i = 0; i < output_batch_dimensions.size(); ++i) in ApplyDelegate()115 output_batch_dim->data[i] = output_batch_dimensions[i]; in ApplyDelegate()120 options.output_batch_dimensions = output_batch_dim; in ApplyDelegate()
47 void SetOutputBatchDimensions(TfLiteIntArray* output_batch_dimensions) { in SetOutputBatchDimensions() argument48 output_batch_dims_ = output_batch_dimensions; in SetOutputBatchDimensions()
59 TfLiteIntArray* output_batch_dimensions) { in CreateBatchSeqBuilder() argument63 builder->SetOutputBatchDimensions(output_batch_dimensions); in CreateBatchSeqBuilder()
54 TfLiteIntArray* output_batch_dimensions);
409 TfLiteIntArray* output_batch_dimensions) { in AddBatchSeqConfig() argument412 input_batch_dimensions, output_batch_dimensions); in AddBatchSeqConfig()
267 TfLiteIntArray* output_batch_dimensions);
90 TfLiteIntArray* output_batch_dimensions; member
179 if (params_.output_batch_dimensions->data[i] == -1) continue; in ResizeOutputTensors()183 new_shape->data[params_.output_batch_dimensions->data[i]] = new_batch; in ResizeOutputTensors()247 params_.output_batch_dimensions); in BuildGraph()