Searched refs:SetInputShape (Results 1 – 4 of 4) 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::__anon1a0dede80111::AddModelBuilder
106 builder.SetInputShape(0, {1, 2, 2, 2}); in Add2SameShapeTensors()
107 builder.SetInputShape(1, {1, 2, 2, 2}); in Add2SameShapeTensors()
114 builder.SetInputShape(0, {1, 2, 2, 2}); in AddBroadcast()
115 builder.SetInputShape(1, {1, 1, 1, 2}); in AddBroadcast()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/ |
| D | api.h | 261 virtual absl::Status SetInputShape(int index,
|
| /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::__anon64adbc2c0111::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::__anon409f18760111::InferenceBuilderImpl
|