Home
last modified time | relevance | path

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

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