Home
last modified time | relevance | path

Searched defs:SetOutputShape (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/lite/delegates/gpu/common/testing/feature_parity/generators/
Dadd.cc94 void SetOutputShape(std::vector<int32_t>&& shape) { in SetOutputShape() function in tflite::__anon51f9a25e0111::AddModelBuilder
/external/tensorflow/tensorflow/lite/kernels/shim/
Dtf_op_shim.cc125 absl::Status TfShapeInferenceContext::SetOutputShape(const int idx, in SetOutputShape() function in tflite::shim::TfShapeInferenceContext
Dop_kernel.h112 absl::Status SetOutputShape(const int idx, const Shape& shape) { in SetOutputShape() function
Dtflite_op_shim.cc141 absl::Status TfLiteShapeInferenceContext::SetOutputShape(const int idx, in SetOutputShape() function in tflite::shim::TfLiteShapeInferenceContext
/external/tensorflow/tensorflow/lite/delegates/coreml/builders/
Dconvolution_op_builder.cc43 void ConvolutionOpBuilder::SetOutputShape(TfLiteTensor* output_shape) { in SetOutputShape() function in tflite::delegates::coreml::ConvolutionOpBuilder