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::__anon3f938d0d0111::AddModelBuilder |
| /external/tensorflow/tensorflow/lite/delegates/gpu/gl/ | ||
| D | api2.cc | 522 absl::Status SetInputShape(int index, const Dimensions& dimensions) final { in SetInputShape() function in tflite::gpu::gl::__anone4be3c310111::InferenceBuilderImpl |
| /external/tensorflow/tensorflow/lite/delegates/gpu/cl/ | ||
| D | api.cc | 741 absl::Status SetInputShape(int index, const Dimensions& dimensions) override { in SetInputShape() function in tflite::gpu::cl::__anonf6ed2bdb0111::InferenceBuilderImpl |