Home
last modified time | relevance | path

Searched defs:GetOutputShape (Results 1 – 25 of 84) sorted by relevance

1234

/external/armnn/src/armnn/test/optimizations/
DFuseActivationTests.cpp51 static TensorShape GetOutputShape() { return TensorShape( {1, 3, 3, 4}); } // NHWCout in GetOutputShape() function
88 static TensorShape GetOutputShape() { return TensorShape( {1, 3, 3, 12}); } // NHWCout in GetOutputShape() function
125 static TensorShape GetOutputShape() { return TensorShape( {2, 3}); } // NCout in GetOutputShape() function
158 static TensorShape GetOutputShape() { return TensorShape( {1, 4, 4, 3}); } // NHWCout in GetOutputShape() function
192 static TensorShape GetOutputShape() { return TensorShape( {1, 4, 4, 3}); } // NHWCout in GetOutputShape() function
212 static TensorShape GetOutputShape() { return TensorShape( {1, 4, 4, 3}); } // NHWCout in GetOutputShape() function
232 static TensorShape GetOutputShape() { return TensorShape( {1, 4, 4, 3}); } // NHWCout in GetOutputShape() function
252 static TensorShape GetOutputShape() { return TensorShape( {1, 4, 4, 3}); } // NHWCout in GetOutputShape() function
/external/tensorflow/tensorflow/lite/kernels/
Dexp_test.cc50 std::vector<int> GetOutputShape() { return GetTensorShape(output_); } in GetOutputShape() function in tflite::__anon4c9d23610111::ExpOpModel
Dfloor_test.cc43 std::vector<int> GetOutputShape() { return GetTensorShape(output_); } in GetOutputShape() function in tflite::__anon64189a560111::FloorOpModel
Dround_test.cc42 std::vector<int> GetOutputShape() { return GetTensorShape(output_); } in GetOutputShape() function in tflite::__anon6cbaa4dc0111::RoundOpModel
Dceil_test.cc43 std::vector<int> GetOutputShape() { return GetTensorShape(output_); } in GetOutputShape() function in tflite::__anon672f6f110111::CeilOpModel
Dfill_test.cc67 std::vector<int> GetOutputShape() { return GetTensorShape(output_); } in GetOutputShape() function in tflite::__anon1b9c0f9b0111::FillOpModel
102 std::vector<int> GetOutputShape() { return GetTensorShape(output_); } in GetOutputShape() function in tflite::__anon1b9c0f9b0111::QuantizedFillOpModel
Dmatrix_diag_test.cc47 std::vector<int> GetOutputShape() { return GetTensorShape(output_); } in GetOutputShape() function in tflite::__anon3d7ccd9d0111::MatrixDiagOpModel
Dmatrix_set_diag_test.cc51 std::vector<int> GetOutputShape() { return GetTensorShape(output_); } in GetOutputShape() function in tflite::__anon9dd7b5280111::MatrixSetDiagOpModel
Dreshape_test_common.h78 std::vector<int> GetOutputShape() { return this->GetTensorShape(output_); } in GetOutputShape() function
Drank_test.cc49 std::vector<int> GetOutputShape() { return GetTensorShape(output_); } in GetOutputShape() function in tflite::__anoncfb777600111::RankOpModel
Dshape_test.cc50 std::vector<int> GetOutputShape() { return GetTensorShape(output_); } in GetOutputShape() function in tflite::__anon76c58ae50111::ShapeOpModel
Dlogical_test.cc44 std::vector<int> GetOutputShape() { return GetTensorShape(output_); } in GetOutputShape() function in tflite::__anonea873d4f0111::LogicalOpModel
Dmfcc_test.cc60 std::vector<int> GetOutputShape() { return GetTensorShape(output_); } in GetOutputShape() function in tflite::ops::custom::__anona56e6a4d0111::BaseMfccOpModel
Daudio_spectrogram_test.cc57 std::vector<int> GetOutputShape() { return GetTensorShape(output_); } in GetOutputShape() function in tflite::ops::custom::__anon0b5d402c0111::BaseAudioSpectrogramOpModel
Dspace_to_depth_test.cc50 std::vector<int> GetOutputShape() { return GetTensorShape(output_); } in GetOutputShape() function in tflite::__anonee463d160111::SpaceToDepthOpModel
Dfake_quant_test.cc49 std::vector<int> GetOutputShape() { return GetTensorShape(output_); } in GetOutputShape() function in tflite::__anon6da287b30111::FakeQuantOpModel
Drfft2d_test.cc53 std::vector<int> GetOutputShape() { return GetTensorShape(output_); } in GetOutputShape() function in tflite::ops::builtin::__anonf8add73c0111::Rfft2dOpModel
Dexpand_dims_test.cc65 std::vector<int> GetOutputShape() { return GetTensorShape(output_); } in GetOutputShape() function in tflite::__anon356c5e600111::ExpandDimsOpModel
Dirfft2d_test.cc53 std::vector<int> GetOutputShape() { return GetTensorShape(output_); } in GetOutputShape() function in tflite::ops::custom::__anon498c5b850111::Irfft2dOpModel
Dbatch_matmul_test.cc73 std::vector<int32_t> GetOutputShape() { return GetTensorShape(output_id_); } in GetOutputShape() function in tflite::__anon5c3be9a50111::BatchMatMulOpModel
343 std::vector<int32_t> GetOutputShape() { in GetOutputShape() function in tflite::__anon5c3be9a50111::ConstRHSBatchMatMulOpModel
411 std::vector<int> GetOutputShape() { return GetTensorShape(output_id_); } in GetOutputShape() function in tflite::__anon5c3be9a50111::HybridBatchMatMulOpModel
Dfloor_div_test.cc46 std::vector<int> GetOutputShape() { return GetTensorShape(output_); } in GetOutputShape() function in tflite::__anon61d085380111::FloorDivModel
/external/tensorflow/tensorflow/lite/delegates/hexagon/builders/tests/
Dspace_to_depth_test.cc49 std::vector<int> GetOutputShape() { return GetTensorShape(output_); } in GetOutputShape() function in tflite::SpaceToDepthOpModel
Dpad_test.cc45 std::vector<int> GetOutputShape() { return GetTensorShape(output_); } in GetOutputShape() function in tflite::PadOpConstModel
Dmul_test.cc51 std::vector<int> GetOutputShape() { return GetTensorShape(output_); } in GetOutputShape() function in tflite::MulOpModel
/external/tensorflow/tensorflow/lite/experimental/acceleration/mini_benchmark/
Ddecode_jpeg_test.cc63 std::vector<int> GetOutputShape() { return GetTensorShape(output_id_); } in GetOutputShape() function in tflite::acceleration::decode_jpeg_kernel::__anonaab57c360111::DecodeJPEGOpModel

1234