Searched defs:SetOutputShape (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/testing/feature_parity/generators/ | ||
| D | add.cc | 94 void SetOutputShape(std::vector<int32_t>&& shape) { in SetOutputShape() function in tflite::__anon5ac0d3af0111::AddModelBuilder |
| /external/tensorflow/tensorflow/lite/delegates/coreml/builders/ | ||
| D | convolution_op_builder.cc | 39 void ConvolutionOpBuilder::SetOutputShape(TfLiteTensor* output_shape) { in SetOutputShape() function in tflite::delegates::coreml::ConvolutionOpBuilder |