Home
last modified time | relevance | path

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

1234

/external/ComputeLibrary/tests/validation/fixtures/
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()
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()
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()
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()
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()
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()
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()
DDirectConvolution3DFixture.h85 …t TensorShape &input_shape, const TensorShape &weights_shape, const TensorShape &bias_shape, const… in compute_target()
130 …t TensorShape &input_shape, const TensorShape &weights_shape, const TensorShape &bias_shape, const… in compute_reference()
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()
/external/ComputeLibrary/tests/validation/fixtures/dynamic_fusion/gpu/cl/
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()
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()
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/sparse_ops/
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()
/external/tensorflow/tensorflow/lite/kernels/internal/reference/integer_ops/
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()
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()
/external/ComputeLibrary/tests/validation/fixtures/UNIT/
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()
/external/tensorflow/tensorflow/lite/kernels/internal/reference/
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()
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()
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/integer_ops/
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()
Dfully_connected.h35 const int8* filter_data, const RuntimeShape& bias_shape, in FullyConnectedPerChannel()
104 const int8* filter_data, const RuntimeShape& bias_shape, in FullyConnected()
/external/ComputeLibrary/tests/validation/CL/
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
/external/ComputeLibrary/tests/validation/NEON/
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
/external/tensorflow/tensorflow/lite/kernels/internal/
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()
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dbias_ops.cc44 const TensorShape bias_shape = ctx->InputShape(1); in Compile() local
/external/tensorflow/tensorflow/lite/kernels/internal/reference/sparse_ops/
Dfully_connected.h29 const RuntimeShape& bias_shape, const float* bias_data, in FullyConnectedSparseWeight()
/external/tensorflow/tensorflow/lite/toco/graph_transformations/
Dfuse_binary_into_preceding_affine.cc57 const Shape& bias_shape = bias.shape(); in FuseAddOrSubParamsIntoPrecedingAffine() local
127 const Shape& bias_shape = bias.shape(); in FuseMulOrDivParamsIntoPrecedingAffine() local

1234