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