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::__anon03bb97c90111::ExpOpModel
Dround_test.cc42 std::vector<int> GetOutputShape() { return GetTensorShape(output_); } in GetOutputShape() function in tflite::__anon6547d3440111::RoundOpModel
Dceil_test.cc43 std::vector<int> GetOutputShape() { return GetTensorShape(output_); } in GetOutputShape() function in tflite::__anon021c70790111::CeilOpModel
Dfloor_test.cc43 std::vector<int> GetOutputShape() { return GetTensorShape(output_); } in GetOutputShape() function in tflite::__anon5ca5c8be0111::FloorOpModel
Dfill_test.cc67 std::vector<int> GetOutputShape() { return GetTensorShape(output_); } in GetOutputShape() function in tflite::__anonb68911030111::FillOpModel
102 std::vector<int> GetOutputShape() { return GetTensorShape(output_); } in GetOutputShape() function in tflite::__anonb68911030111::QuantizedFillOpModel
Dbucketize_test.cc50 std::vector<int> GetOutputShape() { return GetTensorShape(output_); } in GetOutputShape() function in tflite::__anonce1e42c20111::BucketizeOpModel
Dreduce_test_common.h44 std::vector<int> GetOutputShape() { return GetTensorShape(output_); } in GetOutputShape() function
Dmatrix_diag_test.cc47 std::vector<int> GetOutputShape() { return GetTensorShape(output_); } in GetOutputShape() function in tflite::__anone85d2a050111::MatrixDiagOpModel
Dmatrix_set_diag_test.cc51 std::vector<int> GetOutputShape() { return GetTensorShape(output_); } in GetOutputShape() function in tflite::__anon40e005900111::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::__anon6aa478c80111::RankOpModel
Dshape_test.cc50 std::vector<int> GetOutputShape() { return GetTensorShape(output_); } in GetOutputShape() function in tflite::__anon6f52b94d0111::ShapeOpModel
Dlogical_test.cc44 std::vector<int> GetOutputShape() { return GetTensorShape(output_); } in GetOutputShape() function in tflite::__anon3b19a5b70111::LogicalOpModel
Dmfcc_test.cc60 std::vector<int> GetOutputShape() { return GetTensorShape(output_); } in GetOutputShape() function in tflite::ops::custom::__anon405b6bb50111::BaseMfccOpModel
Daudio_spectrogram_test.cc57 std::vector<int> GetOutputShape() { return GetTensorShape(output_); } in GetOutputShape() function in tflite::ops::custom::__anon91bb4a940111::BaseAudioSpectrogramOpModel
Dunsorted_segment_test.h68 std::vector<int32_t> GetOutputShape() { return GetTensorShape(output_id_); } in GetOutputShape() function
Dfake_quant_test.cc49 std::vector<int> GetOutputShape() { return GetTensorShape(output_); } in GetOutputShape() function in tflite::__anon0634f71b0111::FakeQuantOpModel
Dspace_to_depth_test.cc50 std::vector<int> GetOutputShape() { return GetTensorShape(output_); } in GetOutputShape() function in tflite::__anon5fd2207e0111::SpaceToDepthOpModel
Dexpand_dims_test.cc65 std::vector<int> GetOutputShape() { return GetTensorShape(output_); } in GetOutputShape() function in tflite::__anone04cbac80111::ExpandDimsOpModel
/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::__anon5dad059e0111::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