Searched defs:GetInputShape (Results 1 – 6 of 6) sorted by relevance
| /aosp_15_r20/external/armnn/src/armnn/test/optimizations/ |
| H A D | FuseActivationTests.cpp | 46 static TensorShape GetInputShape() { return TensorShape( {1, 4, 4, 3}); } // NHWCin in GetInputShape() function
98 static TensorShape GetInputShape() { return TensorShape( {1, 4, 4, 3}); } // [N,H,W,Cin] in GetInputShape() function
150 static TensorShape GetInputShape() { return TensorShape( {2, 5, 1, 1}); } // NCinHW in GetInputShape() function
198 static TensorShape GetInputShape() { return TensorShape( {1, 4, 4, 3}); } // NHWCin in GetInputShape() function
247 static TensorShape GetInputShape() { return TensorShape( {1, 4, 4, 3}); } // NHWCin in GetInputShape() function
284 static TensorShape GetInputShape() { return TensorShape( {1, 4, 4, 3}); } // NHWCin in GetInputShape() function
321 static TensorShape GetInputShape() { return TensorShape( {1, 4, 4, 3}); } // NHWCin in GetInputShape() function
358 static TensorShape GetInputShape() { return TensorShape( {1, 4, 4, 3}); } // NHWCin in GetInputShape() function
|
| /aosp_15_r20/external/libtextclassifier/native/utils/tflite/ |
| H A D | blacklist.cc | 63 TfLiteIntArray* GetInputShape(TfLiteContext* context, in GetInputShape() function in tflite::ops::custom::libtextclassifier3::blacklist::BlacklistOp
|
| H A D | string_projection.cc | 548 TfLiteIntArray* GetInputShape(TfLiteContext* context, in GetInputShape() function in tflite::ops::custom::libtextclassifier3::string_projection::StringProjectionOp
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/shim/ |
| H A D | tf_op_shim.cc | 115 ShapeOr TfShapeInferenceContext::GetInputShape(const int idx) const { in GetInputShape() function in tflite::shim::TfShapeInferenceContext
|
| H A D | op_kernel.h | 108 ShapeOr GetInputShape(const int idx) const { in GetInputShape() function
|
| H A D | tflite_op_shim.cc | 131 ShapeOr TfLiteShapeInferenceContext::GetInputShape(const int idx) const { in GetInputShape() function in tflite::shim::TfLiteShapeInferenceContext
|