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::__anon51f9a25e0111::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::__anon6db5f6e20111::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::__anoneb8fd46c0111::InferenceBuilderImpl |