Home
last modified time | relevance | path

Searched defs:bias_shape (Results 1 – 25 of 83) sorted by relevance

1234

/aosp_15_r20/external/ComputeLibrary/tests/validation/fixtures/
H A DFullyConnectedLayerFixture.h124 …t TensorShape &input_shape, const TensorShape &weights_shape, const TensorShape &bias_shape, const… in compute_target()
223 …t TensorShape &input_shape, const TensorShape &weights_shape, const TensorShape &bias_shape, const… in compute_reference()
252 …void setup(TensorShape input_shape, TensorShape weights_shape, TensorShape bias_shape, TensorShape… in setup()
266 …void setup(TensorShape input_shape, TensorShape weights_shape, TensorShape bias_shape, TensorShape… in setup()
353 …void setup(TensorShape src_shape, TensorShape weights_shape, TensorShape bias_shape, TensorShape d… in setup()
474 …void setup(TensorShape src_shape, TensorShape weights_shape, TensorShape bias_shape, TensorShape d… in setup()
487 …void setup(TensorShape src_shape, TensorShape weights_shape, TensorShape bias_shape, TensorShape d… in setup()
H A DDirectConvolutionLayerFixture.h77 …void setup(TensorShape input_shape, TensorShape weights_shape, TensorShape bias_shape, TensorShape… in setup()
149 …te_target(TensorShape input_shape, TensorShape weights_shape, const TensorShape &bias_shape, Tenso… in compute_target()
206 …t TensorShape &input_shape, const TensorShape &weights_shape, const TensorShape &bias_shape, const… in compute_reference()
260 …void setup(TensorShape input_shape, TensorShape weights_shape, TensorShape bias_shape, TensorShape… in setup()
273 …void setup(TensorShape input_shape, TensorShape weights_shape, TensorShape bias_shape, TensorShape… in setup()
H A DConvolutionLayerFixture.h366 …void setup(TensorShape input_shape, TensorShape weights_shape, TensorShape bias_shape, TensorShape… in setup()
394 …void setup(TensorShape input_shape, TensorShape weights_shape, TensorShape bias_shape, TensorShape… in setup()
407 …void setup(TensorShape input_shape, TensorShape weights_shape, TensorShape bias_shape, TensorShape… in setup()
518 …void setup(TensorShape input_shape, TensorShape weights_shape, TensorShape bias_shape, TensorShape… in setup()
564 …te_target(TensorShape input_shape, TensorShape weights_shape, const TensorShape &bias_shape, Tenso… in compute_target()
593 …t TensorShape &input_shape, const TensorShape &weights_shape, const TensorShape &bias_shape, const… in compute_reference()
H A DQLSTMLayerNormalizationFixture.h48 …void setup(TensorShape input_shape, TensorShape weight_shape, TensorShape bias_shape, DataType dat… in setup()
100 …get(const TensorShape &input_shape, const TensorShape &weight_shape, const TensorShape &bias_shape) in compute_target()
116 …nce(const TensorShape &input_shape, const TensorShape &weight_shape, const TensorShape &bias_shape) in compute_reference()
H A DRNNLayerFixture.h46 … TensorShape weights_shape, TensorShape recurrent_weights_shape, TensorShape bias_shape, TensorSha… in setup()
64 …e &weights_shape, const TensorShape &recurrent_weights_shape, const TensorShape &bias_shape, const… in compute_target()
114 …sorShape &weights_shape, const TensorShape &recurrent_weights_shape, const TensorShape &bias_shape, in compute_reference()
H A DGEMMFixture.h263 …t(const TensorShape &lhs_shape, const TensorShape &rhs_shape, const TensorShape &bias_shape, DataT… in compute_target()
386 …t(const TensorShape &lhs_shape, const TensorShape &rhs_shape, const TensorShape &bias_shape, DataT… in compute_target()
523 …t(const TensorShape &lhs_shape, const TensorShape &rhs_shape, const TensorShape &bias_shape, const… in compute_target()
678 …t(const TensorShape &lhs_shape, const TensorShape &rhs_shape, const TensorShape &bias_shape, const… in compute_target()
837 …t(const TensorShape &lhs_shape, const TensorShape &rhs_shape, const TensorShape &bias_shape, const… in compute_target()
1035 …t(const TensorShape &lhs_shape, const TensorShape &rhs_shape, const TensorShape &bias_shape, const… in compute_target()
1275 …t(const TensorShape &lhs_shape, const TensorShape &rhs_shape, const TensorShape &bias_shape, const… in compute_target()
1455 …t(const TensorShape &lhs_shape, const TensorShape &rhs_shape, const TensorShape &bias_shape, const… in compute_target()
1634 …t(const TensorShape &lhs_shape, const TensorShape &rhs_shape, const TensorShape &bias_shape, const… in compute_target()
1855 …t(const TensorShape &lhs_shape, const TensorShape &rhs_shape, const TensorShape &bias_shape, const… in compute_target()
[all …]
H A DFFTFixture.h186 …te_target(TensorShape input_shape, TensorShape weights_shape, const TensorShape &bias_shape, Tenso… in compute_target()
244 …t TensorShape &input_shape, const TensorShape &weights_shape, const TensorShape &bias_shape, const… in compute_reference()
275 …void setup(TensorShape input_shape, TensorShape weights_shape, TensorShape bias_shape, TensorShape… in setup()
H A DDeconvolutionLayerFixture.h53 …void setup(TensorShape input_shape, TensorShape weights_shape, TensorShape bias_shape, TensorShape… in setup()
153 …te_target(TensorShape input_shape, TensorShape weights_shape, const TensorShape bias_shape, Tensor… in compute_target()
211 …t TensorShape &input_shape, const TensorShape &weights_shape, const TensorShape &bias_shape, const… in compute_reference()
H A DWinogradConvolutionLayerFixture.h59 …void setup(TensorShape input_shape, TensorShape weights_shape, TensorShape bias_shape, TensorShape… in setup()
109 …mpute_target(TensorShape input_shape, TensorShape weights_shape, TensorShape bias_shape, TensorSha… in compute_target()
166 …t TensorShape &input_shape, const TensorShape &weights_shape, const TensorShape &bias_shape, const… in compute_reference()
/aosp_15_r20/external/ComputeLibrary/tests/validation/fixtures/dynamic_fusion/gpu/cl/
H A DDirectConv2dFixture.h92 …void setup(TensorShape input_shape, TensorShape weights_shape, TensorShape bias_shape, TensorShape… in setup()
109 …te_target(TensorShape input_shape, TensorShape weights_shape, const TensorShape &bias_shape, Conv2… in compute_target()
169 …ce(const TensorShape &input_shape, const TensorShape &weights_shape, const TensorShape &bias_shape, in compute_reference()
207 …nsorShape input_shape, TensorShape weights_shape, TensorShape output_shape, TensorShape bias_shape, in setup()
248 …te_target(TensorShape input_shape, TensorShape weights_shape, const TensorShape &bias_shape, Tenso… in compute_target()
320 …t TensorShape &input_shape, const TensorShape &weights_shape, const TensorShape &bias_shape, const… in compute_reference()
H A DDepthwiseConv2dFixture.h87 const TensorShape bias_shape = TensorShape(weights_shape[2]); in setup() local
119 …te_target(TensorShape input_shape, TensorShape weights_shape, const TensorShape &bias_shape, const… in compute_target()
182 …ce(const TensorShape &input_shape, const TensorShape &weights_shape, const TensorShape &bias_shape, in compute_reference()
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/optimized/sparse_ops/
H A Dfully_connected.h37 const RuntimeShape& bias_shape, const float* bias_data, in FullyConnectedSparseWeight()
83 const RuntimeShape& bias_shape, const int32_t* bias_data, in FullyConnectedSparseWeight1x16Impl()
119 const RuntimeShape& bias_shape, const float* bias_data, in FullyConnectedSparseWeight1x4Impl()
190 const RuntimeShape& bias_shape; member
203 const RuntimeShape& bias_shape, const int32_t* bias_data, in FullyConnectedSparseWeight1x16()
227 const RuntimeShape& bias_shape, const float* bias_data, in FullyConnectedSparseWeight1x4()
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/
H A Ddepthwiseconv_quantized_test.cc127 const RuntimeShape& bias_shape, const int32* bias_data, in DispatchDepthwiseConvGeneral()
143 const int8* filter_data, const RuntimeShape& bias_shape, in DispatchDepthwiseConvGeneral()
163 const RuntimeShape& bias_shape, const int32* bias_data, in DispatchDepthwiseConvImpl()
366 const RuntimeShape& bias_shape, const int32* bias_data, in DispatchDepthwiseConvImpl()
533 const RuntimeShape& bias_shape, const int32* bias_data, in DispatchDepthwiseConv()
618 const std::int32_t* bias_data, const RuntimeShape& bias_shape, int stride, in TestOneDepthwiseConvWithGivenOutputShift()
755 const std::int32_t* bias_data, const RuntimeShape& bias_shape, int stride, in TestOneDepthwiseConvBisectOutputShift()
814 const std::int32_t* bias_data, const RuntimeShape& bias_shape, int stride, in TestOneDepthwiseConv()
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/reference/integer_ops/
H A Dfully_connected.h36 const int8_t* filter_data, const RuntimeShape& bias_shape, in FullyConnectedPerChannel()
78 const int8_t* filter_data, const RuntimeShape& bias_shape, in FullyConnectedPerChannel()
116 const int8_t* filter_data, const RuntimeShape& bias_shape, in FullyConnected()
160 const int8_t* filter_data, const RuntimeShape& bias_shape, in FullyConnected()
H A Ddepthwise_conv.h28 const int8_t* filter_data, const RuntimeShape& bias_shape, in DepthwiseConvPerChannel()
128 const int8_t* filter_data, const RuntimeShape& bias_shape, in DepthwiseConvPerChannel()
214 const RuntimeShape& bias_shape, const float* bias_data, in DepthwiseConvHybridPerChannel()
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/optimized/integer_ops/
H A Ddepthwise_conv_hybrid.h46 const int8* filter_data, const RuntimeShape& bias_shape, in DepthwiseConvHybridGeneral()
265 const RuntimeShape& bias_shape, const float* bias_data, in DepthwiseConvHybridWithRounding()
324 const RuntimeShape& bias_shape, const float* bias_data, in DepthwiseConvHybridImpl()
383 const RuntimeShape& bias_shape; member
398 const RuntimeShape& bias_shape, const float* bias_data, in DepthwiseConvHybridPerChannel()
H A Dfully_connected.h35 const int8* filter_data, const RuntimeShape& bias_shape, in FullyConnectedPerChannel()
104 const int8* filter_data, const RuntimeShape& bias_shape, in FullyConnected()
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/reference/
H A Dfully_connected.h32 const float* weights_data, const RuntimeShape& bias_shape, in FullyConnected()
68 const uint8_t* filter_data, const RuntimeShape& bias_shape, in FullyConnected()
116 const uint8_t* filter_data, const RuntimeShape& bias_shape, in FullyConnected()
170 const uint8_t* shuffled_weights_data, const RuntimeShape& bias_shape, in ShuffledFullyConnected()
H A Dsvdf.h70 const int16_t* weights_time_data, const RuntimeShape& bias_shape, in EvalIntegerSVDF()
156 const float* weights_time_data, const RuntimeShape& bias_shape, in EvalFloatSVDF()
196 const RuntimeShape& bias_shape, const float* bias_data, float* scratch, in EvalHybridSVDF()
H A Dconv.h29 const float* filter_data, const RuntimeShape& bias_shape, in Conv()
112 const uint8_t* filter_data, const RuntimeShape& bias_shape, in Conv()
207 const RuntimeShape& bias_shape, const float* bias_data, in HybridConvPerChannel()
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/
H A Dremapper_test.cc639 auto bias_shape = ops::Placeholder::Shape({128}); in RunTest() local
746 auto bias_shape = Placeholder::Shape({128}); in RunTest() local
900 auto bias_shape = ops::Placeholder::Shape({128}); in RunTest() local
1284 auto bias_shape = ops::Placeholder::Shape({128}); in RunTest() local
1427 auto bias_shape = ops::Placeholder::Shape({64}); in RunTest() local
1522 auto bias_shape = Placeholder::Shape({128}); in TEST_F() local
1604 auto bias_shape = ops::Placeholder::Shape({64}); in RunTest() local
1914 auto bias_shape = ops::Placeholder::Shape({128}); in TEST_F() local
1986 auto bias_shape = ops::Placeholder::Shape({128}); in TEST_F() local
2058 auto bias_shape = Placeholder::Shape({128}); in TEST_F() local
[all …]
H A Dmkl_remapper_test.cc61 auto bias_shape = ops::Placeholder::Shape({128}); in FuseConv2DWithBiasAndAddNOrAdd() local
458 auto bias_shape = ops::Placeholder::Shape({8}); in TEST_F() local
549 auto bias_shape = Placeholder::Shape({128 * 3}); in RelpaceAddWithBiasAddDepthwiseConv2D() local
661 auto bias_shape = ops::Placeholder::Shape({64}); in RunTest() local
/aosp_15_r20/external/ComputeLibrary/tests/validation/fixtures/UNIT/
H A DDynamicTensorFixture.h255 …td::vector<TensorShape> input_shapes, TensorShape weights_shape, TensorShape bias_shape, std::vect… in setup()
296 …e run_target(TensorShape input_shape, TensorShape weights_shape, TensorShape bias_shape, TensorSha… in run_target()
345 …un_reference(TensorShape input_shape, TensorShape weights_shape, TensorShape bias_shape, TensorSha… in run_reference()
/aosp_15_r20/external/ComputeLibrary/tests/validation/CL/
H A DLSTMLayerQuantized.cpp91 TensorShape bias_shape{ output_size }; in TEST_SUITE() local
230 TensorShape bias_shape{ output_size }; in TEST_CASE() local
470 TensorShape bias_shape{ output_size }; in TEST_SUITE() local
/aosp_15_r20/external/ComputeLibrary/tests/validation/NEON/
H A DLSTMLayerQuantized.cpp98 TensorShape bias_shape{ output_size }; in TEST_SUITE() local
237 TensorShape bias_shape{ output_size }; in TEST_CASE() local
477 TensorShape bias_shape{ output_size }; in TEST_SUITE() local

1234