Searched refs:SetOutputShape (Results 1 – 3 of 3) sorted by relevance
94 void SetOutputShape(std::vector<int32_t>&& shape) { in SetOutputShape() function in tflite::__anon5ac0d3af0111::AddModelBuilder108 builder.SetOutputShape({1, 2, 2, 2}); in Add2SameShapeTensors()116 builder.SetOutputShape({1, 2, 2, 2}); in AddBroadcast()
50 void SetOutputShape(TfLiteTensor* output_shape);
39 void ConvolutionOpBuilder::SetOutputShape(TfLiteTensor* output_shape) { in SetOutputShape() function in tflite::delegates::coreml::ConvolutionOpBuilder252 SetOutputShape(&context->tensors[inputs->data[0]]); in RegisterInputs()