Home
last modified time | relevance | path

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

123

/external/ComputeLibrary/tests/validation/fixtures/
DFullyConnectedLayerFixture.h58 …void setup(TensorShape input_shape, TensorShape weights_shape, TensorShape bias_shape, TensorShape… in setup()
103 …t TensorShape &input_shape, const TensorShape &weights_shape, const TensorShape &bias_shape, const… in compute_target()
193 …t TensorShape &input_shape, const TensorShape &weights_shape, const TensorShape &bias_shape, const… in compute_reference()
221 …void setup(TensorShape input_shape, TensorShape weights_shape, TensorShape bias_shape, TensorShape… in setup()
235 …void setup(TensorShape input_shape, TensorShape weights_shape, TensorShape bias_shape, TensorShape… in setup()
DFFTFixture.h121 …void setup(TensorShape input_shape, TensorShape weights_shape, TensorShape bias_shape, TensorShape… in setup()
148 …te_target(TensorShape input_shape, TensorShape weights_shape, const TensorShape &bias_shape, Tenso… in compute_target()
198 …t TensorShape &input_shape, const TensorShape &weights_shape, const TensorShape &bias_shape, const… in compute_reference()
228 …void setup(TensorShape input_shape, TensorShape weights_shape, TensorShape bias_shape, TensorShape… 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()
DConvolutionLayerFixture.h71 …void setup(TensorShape input_shape, TensorShape weights_shape, TensorShape bias_shape, TensorShape… in setup()
156 …te_target(TensorShape input_shape, TensorShape weights_shape, const TensorShape &bias_shape, Tenso… in compute_target()
213 …t TensorShape &input_shape, const TensorShape &weights_shape, const TensorShape &bias_shape, const… in compute_reference()
264 …void setup(TensorShape input_shape, TensorShape weights_shape, TensorShape bias_shape, TensorShape… in setup()
278 …void setup(TensorShape input_shape, TensorShape weights_shape, TensorShape bias_shape, TensorShape… in setup()
291 …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()
61 …e &weights_shape, const TensorShape &recurrent_weights_shape, const TensorShape &bias_shape, const… in compute_target()
111 …sorShape &weights_shape, const TensorShape &recurrent_weights_shape, const TensorShape &bias_shape, in compute_reference()
DDirectConvolutionLayerFixture.h76 …void setup(TensorShape input_shape, TensorShape weights_shape, TensorShape bias_shape, TensorShape… in setup()
128 …te_target(TensorShape input_shape, TensorShape weights_shape, const TensorShape &bias_shape, Tenso… in compute_target()
175 …t TensorShape &input_shape, const TensorShape &weights_shape, const TensorShape &bias_shape, const… in compute_reference()
228 …void setup(TensorShape input_shape, TensorShape weights_shape, TensorShape bias_shape, TensorShape… in setup()
241 …void setup(TensorShape input_shape, TensorShape weights_shape, TensorShape bias_shape, TensorShape… in setup()
DDirectConvolutionLayerTensorShiftFixture.h69 …void setup(TensorShape input_shape, TensorShape weights_shape, TensorShape bias_shape, TensorShape… in setup()
113 …t TensorShape &input_shape, const TensorShape &weights_shape, const TensorShape &bias_shape, const… in compute_target()
164 …t TensorShape &input_shape, const TensorShape &weights_shape, const TensorShape &bias_shape, const… in compute_reference()
234 …void setup(TensorShape input_shape, TensorShape weights_shape, TensorShape bias_shape, TensorShape… in setup()
247 …void setup(TensorShape input_shape, TensorShape weights_shape, TensorShape bias_shape, TensorShape… in setup()
DWinogradConvolutionLayerFixture.h59 …void setup(TensorShape input_shape, TensorShape weights_shape, TensorShape bias_shape, TensorShape… in setup()
88 …mpute_target(TensorShape input_shape, TensorShape weights_shape, TensorShape bias_shape, TensorSha… in compute_target()
129 …t TensorShape &input_shape, const TensorShape &weights_shape, const TensorShape &bias_shape, const… in compute_reference()
163 …void setup(TensorShape input_shape, TensorShape weights_shape, TensorShape bias_shape, TensorShape… in setup()
197 …mpute_target(TensorShape input_shape, TensorShape weights_shape, TensorShape bias_shape, TensorSha… in compute_target()
246 …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()
126 …te_target(TensorShape input_shape, TensorShape weights_shape, const TensorShape bias_shape, Tensor… in compute_target()
185 …t TensorShape &input_shape, const TensorShape &weights_shape, const TensorShape &bias_shape, const… in compute_reference()
DGEMMFixture.h202 …t(const TensorShape &lhs_shape, const TensorShape &rhs_shape, const TensorShape &bias_shape, DataT… in compute_target()
315 …t(const TensorShape &lhs_shape, const TensorShape &rhs_shape, const TensorShape &bias_shape, DataT… in compute_target()
442 …t(const TensorShape &lhs_shape, const TensorShape &rhs_shape, const TensorShape &bias_shape, const… in compute_target()
581 …t(const TensorShape &lhs_shape, const TensorShape &rhs_shape, const TensorShape &bias_shape, const… in compute_target()
724 …t(const TensorShape &lhs_shape, const TensorShape &rhs_shape, const TensorShape &bias_shape, const… in compute_target()
889 …t(const TensorShape &lhs_shape, const TensorShape &rhs_shape, const TensorShape &bias_shape, const… in compute_target()
1053 …t(const TensorShape &lhs_shape, const TensorShape &rhs_shape, const TensorShape &bias_shape, const… in compute_target()
1201 …t(const TensorShape &lhs_shape, const TensorShape &rhs_shape, const TensorShape &bias_shape, const… in compute_target()
1352 …t(const TensorShape &lhs_shape, const TensorShape &rhs_shape, const TensorShape &bias_shape, const… in compute_target()
1476 …t(const TensorShape &lhs_shape, const TensorShape &rhs_shape, const TensorShape &bias_shape, const… in compute_target()
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/sparse_ops/
Dfully_connected.h34 const RuntimeShape& bias_shape, const float* bias_data, in FullyConnectedSparseWeight()
80 const RuntimeShape& bias_shape, const float* bias_data, in FullyConnectedSparseWeight1x4Impl()
151 const RuntimeShape& bias_shape; member
168 const RuntimeShape& bias_shape, const float* bias_data, in FullyConnectedSparseWeight1x4()
/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.h29 const float* weights_data, const RuntimeShape& bias_shape, in FullyConnected()
65 const uint8_t* filter_data, const RuntimeShape& bias_shape, in FullyConnected()
113 const uint8_t* filter_data, const RuntimeShape& bias_shape, in FullyConnected()
167 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()
Dconv.h27 const float* filter_data, const RuntimeShape& bias_shape, in Conv()
103 const uint8_t* filter_data, const RuntimeShape& bias_shape, in Conv()
191 const RuntimeShape& bias_shape, const float* bias_data, in HybridConvPerChannel()
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/integer_ops/
Ddepthwise_conv_hybrid.h44 const int8* filter_data, const RuntimeShape& bias_shape, in DepthwiseConvHybridGeneral()
263 const RuntimeShape& bias_shape, const float* bias_data, in DepthwiseConvHybridWithRounding()
322 const RuntimeShape& bias_shape, const float* bias_data, in DepthwiseConvHybridImpl()
381 const RuntimeShape& bias_shape; member
396 const RuntimeShape& bias_shape, const float* bias_data, in DepthwiseConvHybridPerChannel()
/external/tensorflow/tensorflow/lite/kernels/internal/reference/integer_ops/
Dfully_connected.h26 const int8_t* filter_data, const RuntimeShape& bias_shape, in FullyConnected()
68 const int8_t* filter_data, const RuntimeShape& bias_shape, in FullyConnected()
Ddepthwise_conv.h26 const int8_t* filter_data, const RuntimeShape& bias_shape, in DepthwiseConvPerChannel()
126 const int8_t* filter_data, const RuntimeShape& bias_shape, in DepthwiseConvPerChannel()
212 const RuntimeShape& bias_shape, const float* bias_data, in DepthwiseConvHybridPerChannel()
Dconv.h28 const int8_t* filter_data, const RuntimeShape& bias_shape, in ConvPerChannel()
133 const int8_t* filter_data, const RuntimeShape& bias_shape, in ConvPerChannel()
/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/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/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

123