Home
last modified time | relevance | path

Searched defs:SetInputShape (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/lite/delegates/gpu/common/testing/feature_parity/generators/
Dadd.cc86 void SetInputShape(uint32_t input, std::vector<int32_t>&& shape) { in SetInputShape() function in tflite::__anon3f938d0d0111::AddModelBuilder
/external/tensorflow/tensorflow/lite/delegates/gpu/gl/
Dapi2.cc522 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/
Dapi.cc741 absl::Status SetInputShape(int index, const Dimensions& dimensions) override { in SetInputShape() function in tflite::gpu::cl::__anonf6ed2bdb0111::InferenceBuilderImpl