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::__anon5ac0d3af0111::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::__anon517d4d930111::InferenceBuilderImpl |
| /external/tensorflow/tensorflow/lite/delegates/gpu/cl/ | ||
| D | api.cc | 726 absl::Status SetInputShape(int index, const Dimensions& dimensions) override { in SetInputShape() function in tflite::gpu::cl::__anon117e98fd0111::InferenceBuilderImpl |