Home
last modified time | relevance | path

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

1234

/external/armnn/src/armnn/test/optimizations/
DFuseActivationTests.cpp47 static TensorShape GetOutputShape() { return TensorShape( {1, 3, 3, 4}); } // NHWCout in GetOutputShape() function
99 static TensorShape GetOutputShape() { return TensorShape( {1, 3, 3, 12}); } // [N,H,W,Cout] in GetOutputShape() function
151 static TensorShape GetOutputShape() { return TensorShape( {2, 3}); } // NCout in GetOutputShape() function
199 static TensorShape GetOutputShape() { return TensorShape( {1, 4, 4, 3}); } // NHWCout in GetOutputShape() function
248 static TensorShape GetOutputShape() { return TensorShape( {1, 4, 4, 3}); } // NHWCout in GetOutputShape() function
285 static TensorShape GetOutputShape() { return TensorShape( {1, 4, 4, 3}); } // NHWCout in GetOutputShape() function
322 static TensorShape GetOutputShape() { return TensorShape( {1, 4, 4, 3}); } // NHWCout in GetOutputShape() function
359 static TensorShape GetOutputShape() { return TensorShape( {1, 4, 4, 3}); } // NHWCout in GetOutputShape() function
/external/tensorflow/tensorflow/lite/kernels/
Dfloor_mod_test_common.h38 std::vector<int> GetOutputShape() { return GetTensorShape(output_); } in GetOutputShape() function
Dexp_test.cc50 std::vector<int> GetOutputShape() { return GetTensorShape(output_); } in GetOutputShape() function in tflite::__anon492ab29a0111::ExpOpModel
Dfloor_test.cc43 std::vector<int> GetOutputShape() { return GetTensorShape(output_); } in GetOutputShape() function in tflite::__anonba46dbcf0111::FloorOpModel
Dround_test.cc42 std::vector<int> GetOutputShape() { return GetTensorShape(output_); } in GetOutputShape() function in tflite::__anonc2e8e6550111::RoundOpModel
Dceil_test.cc43 std::vector<int> GetOutputShape() { return GetTensorShape(output_); } in GetOutputShape() function in tflite::__anonf56ee56a0111::CeilOpModel
Dfill_test.cc67 std::vector<int> GetOutputShape() { return GetTensorShape(output_); } in GetOutputShape() function in tflite::__anona9db85f40111::FillOpModel
102 std::vector<int> GetOutputShape() { return GetTensorShape(output_); } in GetOutputShape() function in tflite::__anona9db85f40111::QuantizedFillOpModel
Dreduce_test_common.h44 std::vector<int> GetOutputShape() { return GetTensorShape(output_); } in GetOutputShape() function
Dbucketize_test.cc50 std::vector<int> GetOutputShape() { return GetTensorShape(output_); } in GetOutputShape() function in tflite::__anon6d4476530111::BucketizeOpModel
Dmatrix_diag_test.cc47 std::vector<int> GetOutputShape() { return GetTensorShape(output_); } in GetOutputShape() function in tflite::__anone9de85d60111::MatrixDiagOpModel
Dmatrix_set_diag_test.cc51 std::vector<int> GetOutputShape() { return GetTensorShape(output_); } in GetOutputShape() function in tflite::__anon985ce1e10111::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::__anon5df6edb90111::RankOpModel
Dshape_test.cc50 std::vector<int> GetOutputShape() { return GetTensorShape(output_); } in GetOutputShape() function in tflite::__anonccf3cc5e0111::ShapeOpModel
Dlogical_test.cc44 std::vector<int> GetOutputShape() { return GetTensorShape(output_); } in GetOutputShape() function in tflite::__anon854bc1080111::LogicalOpModel
Dmfcc_test.cc60 std::vector<int> GetOutputShape() { return GetTensorShape(output_); } in GetOutputShape() function in tflite::ops::custom::__anon33ade0a60111::BaseMfccOpModel
Daudio_spectrogram_test.cc57 std::vector<int> GetOutputShape() { return GetTensorShape(output_); } in GetOutputShape() function in tflite::ops::custom::__anonbbe07f250111::BaseAudioSpectrogramOpModel
Dunsorted_segment_test.h68 std::vector<int32_t> GetOutputShape() { return GetTensorShape(output_id_); } in GetOutputShape() function
Dspace_to_depth_test.cc50 std::vector<int> GetOutputShape() { return GetTensorShape(output_); } in GetOutputShape() function in tflite::__anonb00c36af0111::SpaceToDepthOpModel
Dfake_quant_test.cc49 std::vector<int> GetOutputShape() { return GetTensorShape(output_); } in GetOutputShape() function in tflite::__anon8a219ccc0111::FakeQuantOpModel
Drfft2d_test.cc53 std::vector<int> GetOutputShape() { return GetTensorShape(output_); } in GetOutputShape() function in tflite::ops::builtin::__anon14a447d50111::Rfft2dOpModel
/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.cc47 std::vector<int> GetOutputShape() { return GetTensorShape(output_); } in GetOutputShape() function in tflite::PadOpConstModel
/external/tensorflow/tensorflow/lite/experimental/acceleration/mini_benchmark/
Ddecode_jpeg_test.cc64 std::vector<int> GetOutputShape() { return GetTensorShape(output_id_); } in GetOutputShape() function in tflite::acceleration::decode_jpeg_kernel::__anondaed5f8f0111::DecodeJPEGOpModel
/external/tflite-support/tensorflow_lite_support/custom_ops/kernel/
Dunsorted_segment_test.h68 std::vector<int32_t> GetOutputShape() { return GetTensorShape(output_id_); } in GetOutputShape() function

1234