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::__anon0e55baab0111::ExpOpModel
Dfloor_test.cc43 std::vector<int> GetOutputShape() { return GetTensorShape(output_); } in GetOutputShape() function in tflite::__anon76542c200111::FloorOpModel
Dceil_test.cc43 std::vector<int> GetOutputShape() { return GetTensorShape(output_); } in GetOutputShape() function in tflite::__anon5ffaef9b0111::CeilOpModel
Dround_test.cc42 std::vector<int> GetOutputShape() { return GetTensorShape(output_); } in GetOutputShape() function in tflite::__anon7ef636a60111::RoundOpModel
Dfill_test.cc67 std::vector<int> GetOutputShape() { return GetTensorShape(output_); } in GetOutputShape() function in tflite::__anon146790250111::FillOpModel
102 std::vector<int> GetOutputShape() { return GetTensorShape(output_); } in GetOutputShape() function in tflite::__anon146790250111::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::__anon153387240111::BucketizeOpModel
Dmatrix_diag_test.cc47 std::vector<int> GetOutputShape() { return GetTensorShape(output_); } in GetOutputShape() function in tflite::__anon49d50ee70111::MatrixDiagOpModel
Dmatrix_set_diag_test.cc51 std::vector<int> GetOutputShape() { return GetTensorShape(output_); } in GetOutputShape() function in tflite::__anonb3a38b720111::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::__anonc882f7ea0111::RankOpModel
Dshape_test.cc50 std::vector<int> GetOutputShape() { return GetTensorShape(output_); } in GetOutputShape() function in tflite::__anon89011caf0111::ShapeOpModel
Dmfcc_test.cc60 std::vector<int> GetOutputShape() { return GetTensorShape(output_); } in GetOutputShape() function in tflite::ops::custom::__anon9e39ead70111::BaseMfccOpModel
Dlogical_test.cc44 std::vector<int> GetOutputShape() { return GetTensorShape(output_); } in GetOutputShape() function in tflite::__anon79ee69990111::LogicalOpModel
Daudio_spectrogram_test.cc57 std::vector<int> GetOutputShape() { return GetTensorShape(output_); } in GetOutputShape() function in tflite::ops::custom::__anonc377d0f60111::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::__anondf6b72200111::SpaceToDepthOpModel
Dfake_quant_test.cc49 std::vector<int> GetOutputShape() { return GetTensorShape(output_); } in GetOutputShape() function in tflite::__anon2ff2c7bd0111::FakeQuantOpModel
Dexpand_dims_test.cc65 std::vector<int> GetOutputShape() { return GetTensorShape(output_); } in GetOutputShape() function in tflite::__anon41c49faa0111::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/tflite-support/tensorflow_lite_support/custom_ops/kernel/
Dunsorted_segment_test.h68 std::vector<int32_t> GetOutputShape() { return GetTensorShape(output_id_); } in GetOutputShape() function
/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::__anon214d8ec00111::DecodeJPEGOpModel

1234