Searched defs:SetInputShape (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/common/testing/feature_parity/generators/ | ||
| H A D | add.cc | 86 void SetInputShape(uint32_t input, std::vector<int32_t>&& shape) { in SetInputShape() function in tflite::__anond1ed37de0111::AddModelBuilder |
| /aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/gl/ | ||
| H A D | api2.cc | 522 absl::Status SetInputShape(int index, const Dimensions& dimensions) final { in SetInputShape() function in tflite::gpu::gl::__anon11001c620111::InferenceBuilderImpl |
| /aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/cl/ | ||
| H A D | api.cc | 741 absl::Status SetInputShape(int index, const Dimensions& dimensions) override { in SetInputShape() function in tflite::gpu::cl::__anon936b49ec0111::InferenceBuilderImpl |