Home
last modified time | relevance | path

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

1234

/aosp_15_r20/external/armnn/src/armnn/test/optimizations/
H A 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
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/
H A Dfill_test.cc67 std::vector<int> GetOutputShape() { return GetTensorShape(output_); } in GetOutputShape() function in tflite::__anon397af5a50111::FillOpModel
102 std::vector<int> GetOutputShape() { return GetTensorShape(output_); } in GetOutputShape() function in tflite::__anon397af5a50111::QuantizedFillOpModel
H A Dbatch_matmul_test.cc73 std::vector<int32_t> GetOutputShape() { return GetTensorShape(output_id_); } in GetOutputShape() function in tflite::__anond27d39af0111::BatchMatMulOpModel
356 std::vector<int32_t> GetOutputShape() { in GetOutputShape() function in tflite::__anond27d39af0111::ConstRHSBatchMatMulOpModel
425 std::vector<int> GetOutputShape() { return GetTensorShape(output_id_); } in GetOutputShape() function in tflite::__anond27d39af0111::HybridBatchMatMulOpModel
H A Dtranspose_conv_test.cc108 std::vector<int> GetOutputShape() { return GetTensorShape(output_); } in GetOutputShape() function in tflite::__anonf3f7b7120111::BaseTransposeConvOpModel
668 std::vector<int> GetOutputShape() { return GetTensorShape(output_); } in GetOutputShape() function in tflite::__anonf3f7b7120111::BaseTransposeConvBiasOpModel
H A Dfloor_mod_test_common.h38 std::vector<int> GetOutputShape() { return GetTensorShape(output_); } in GetOutputShape() function
H A Dexp_test.cc50 std::vector<int> GetOutputShape() { return GetTensorShape(output_); } in GetOutputShape() function in tflite::__anon83d2702b0111::ExpOpModel
H A Dfloor_test.cc43 std::vector<int> GetOutputShape() { return GetTensorShape(output_); } in GetOutputShape() function in tflite::__anon3dd441a00111::FloorOpModel
H A Dround_test.cc42 std::vector<int> GetOutputShape() { return GetTensorShape(output_); } in GetOutputShape() function in tflite::__anon46764c260111::RoundOpModel
H A Dceil_test.cc43 std::vector<int> GetOutputShape() { return GetTensorShape(output_); } in GetOutputShape() function in tflite::__anon850e551b0111::CeilOpModel
H A Dbucketize_test.cc50 std::vector<int> GetOutputShape() { return GetTensorShape(output_); } in GetOutputShape() function in tflite::__anon21c25ca40111::BucketizeOpModel
H A Dreduce_test_common.h44 std::vector<int> GetOutputShape() { return GetTensorShape(output_); } in GetOutputShape() function
H A Dmatrix_diag_test.cc47 std::vector<int> GetOutputShape() { return GetTensorShape(output_); } in GetOutputShape() function in tflite::__anonb56f44670111::MatrixDiagOpModel
H A Dmatrix_set_diag_test.cc51 std::vector<int> GetOutputShape() { return GetTensorShape(output_); } in GetOutputShape() function in tflite::__anon245c80f20111::MatrixSetDiagOpModel
H A Drank_test.cc49 std::vector<int> GetOutputShape() { return GetTensorShape(output_); } in GetOutputShape() function in tflite::__anoned965d6a0111::RankOpModel
H A Dreshape_test_common.h78 std::vector<int> GetOutputShape() { return this->GetTensorShape(output_); } in GetOutputShape() function
H A Dshape_test.cc50 std::vector<int> GetOutputShape() { return GetTensorShape(output_); } in GetOutputShape() function in tflite::__anon5081322f0111::ShapeOpModel
H A Dlogical_test.cc44 std::vector<int> GetOutputShape() { return GetTensorShape(output_); } in GetOutputShape() function in tflite::__anon21c9df190111::LogicalOpModel
H A Dmfcc_test.cc60 std::vector<int> GetOutputShape() { return GetTensorShape(output_); } in GetOutputShape() function in tflite::ops::custom::__anonc34d50570111::BaseMfccOpModel
H A Dfully_connected_test.cc276 std::vector<int> GetOutputShape() { return GetTensorShape(output_); } in GetOutputShape() function in tflite::__anonaa112c3c0111::FloatFullyConnectedOpModel
449 std::vector<int> GetOutputShape() { return GetTensorShape(output_); } in GetOutputShape() function in tflite::__anonaa112c3c0111::HybridFullyConnectedOpModel
1594 std::vector<int> GetOutputShape() { return GetTensorShape(output_); } in GetOutputShape() function in tflite::__anonaa112c3c0111::SparseFullyConnectedOpModel
H A Dunsorted_segment_test.h68 std::vector<int32_t> GetOutputShape() { return GetTensorShape(output_id_); } in GetOutputShape() function
H A Daudio_spectrogram_test.cc57 std::vector<int> GetOutputShape() { return GetTensorShape(output_); } in GetOutputShape() function in tflite::ops::custom::__anon464426760111::BaseAudioSpectrogramOpModel
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/nnapi/
H A Dnnapi_delegate_test.cc1512 std::vector<int> GetOutputShape() { return GetTensorShape(output_); } in GetOutputShape() function in tflite::__anon1a74836d0111::ReshapeOpModel
1545 std::vector<int> GetOutputShape() { return GetTensorShape(output_); } in GetOutputShape() function in tflite::__anon1a74836d0111::SqueezeOpModel
1601 std::vector<int> GetOutputShape() { return GetTensorShape(output_); } in GetOutputShape() function in tflite::__anon1a74836d0111::L2NormOpModel
1639 std::vector<int> GetOutputShape() { return GetTensorShape(output_); } in GetOutputShape() function in tflite::__anon1a74836d0111::TransposeSimpleModel
1996 std::vector<int> GetOutputShape() { return GetTensorShape(output_); } in GetOutputShape() function in tflite::__anon1a74836d0111::FloorOpModel
2458 std::vector<int> GetOutputShape() { return GetTensorShape(output_); } in GetOutputShape() function in tflite::__anon1a74836d0111::PadOpModel
2516 std::vector<int> GetOutputShape() { return GetTensorShape(output_); } in GetOutputShape() function in tflite::__anon1a74836d0111::SpaceToBatchNDOpModel
2617 std::vector<int> GetOutputShape() { return GetTensorShape(output_); } in GetOutputShape() function in tflite::__anon1a74836d0111::StridedSliceOpModel
4748 std::vector<int> GetOutputShape() { return GetTensorShape(output_); } in GetOutputShape() function in tflite::__anon1a74836d0111::BaseReduceOpModel
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/hexagon/builders/tests/
H A Dspace_to_depth_test.cc49 std::vector<int> GetOutputShape() { return GetTensorShape(output_); } in GetOutputShape() function in tflite::SpaceToDepthOpModel
/aosp_15_r20/external/tensorflow/tensorflow/lite/experimental/acceleration/mini_benchmark/
H A Ddecode_jpeg_test.cc64 std::vector<int> GetOutputShape() { return GetTensorShape(output_id_); } in GetOutputShape() function in tflite::acceleration::decode_jpeg_kernel::__anonfe6474400111::DecodeJPEGOpModel
/aosp_15_r20/external/tflite-support/tensorflow_lite_support/custom_ops/kernel/
H A Dunsorted_segment_test.h68 std::vector<int32_t> GetOutputShape() { return GetTensorShape(output_id_); } in GetOutputShape() function

1234