Searched refs:BiasType (Results 1 – 11 of 11) sorted by relevance
| /external/tensorflow/tensorflow/lite/tools/optimize/ |
| D | quantization_utils.cc | 612 template <class BiasType>
613 std::vector<BiasType> SymmetricBiasQuantize(const float* data, in SymmetricBiasQuantize()
616 std::vector<BiasType> buffer(num_elements); in SymmetricBiasQuantize()
617 const BiasType kScale = std::numeric_limits<BiasType>::max(); in SymmetricBiasQuantize()
624 const BiasType quantized_value = in SymmetricBiasQuantize()
625 tflite::SafeCast<BiasType>(TfLiteRound(data[idx] * scaling_factor_inv)); in SymmetricBiasQuantize()
634 template <class BiasType>
643 auto final_buffer = SymmetricBiasQuantize<BiasType>(float_data, num_elements, in SymmetricPerLayerBiasQuantize()
648 size_t buffer_size = num_elements * sizeof(BiasType); in SymmetricPerLayerBiasQuantize()
652 auto output_type = std::is_same<BiasType, std::int32_t>::value in SymmetricPerLayerBiasQuantize()
[all …]
|
| D | quantization_utils.h | 121 template <typename BiasType>
128 template <typename BiasType>
135 template <typename BiasType>
136 std::vector<BiasType> SymmetricBiasQuantize(const float* data,
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_conv_runner.cc | 115 template <typename ElementType, typename BiasType, typename OutputType>
185 template <typename ElementType, typename BiasType, typename OutputType,
201 return RunGpuConvForwardActivation<ElementType, BiasType, OutputType>( in RunGpuConvInternalImpl()
210 template <typename ElementType, typename BiasType, typename OutputType,
224 return RunGpuConvForwardActivation<ElementType, BiasType, OutputType>( in RunGpuConvInternalImpl()
235 template <typename ElementType, typename BiasType, typename OutputType>
248 Status run_status = RunGpuConvInternalImpl<ElementType, BiasType, OutputType>( in RunGpuConvImpl()
|
| /external/ComputeLibrary/tests/validation/fixtures/ |
| D | QLSTMLayerNormalizationFixture.h | 60 template <typename InputType, typename BiasType>
61 void fill(InputType &&input_tensor, InputType &&weight_tensor, BiasType &&bias_tensor) in fill()
|
| /external/gemmlowp/meta/ |
| D | transform_kernels.h | 54 template <typename BiasType>
67 const BiasType* bias;
|
| /external/XNNPACK/test/ |
| D | fully-connected.cc | 24 template <class T, class BiasType = T> class FullyConnectedTestBase : public ::testing::Test {
59 bias = std::vector<BiasType>(output_channels); in FullyConnectedTestBase()
104 std::vector<BiasType> bias;
|
| D | deconvolution-2d.cc | 23 template <class T, class BiasType = T> class DeconvolutionTestBase : public ::testing::Test {
65 bias = std::vector<BiasType>(groups * group_output_channels); in DeconvolutionTestBase()
109 std::vector<BiasType> bias;
|
| D | unpooling-2d.cc | 23 template <class T, class BiasType = T> class Unpooling2DTestBase : public ::testing::Test {
|
| D | convolution-2d.cc | 25 template <class T, class BiasType = T> class ConvolutionTestBase : public ::testing::Test {
63 bias = std::vector<BiasType>(groups * group_output_channels); in ConvolutionTestBase()
105 std::vector<BiasType> bias;
|
| D | depthwise-convolution-2d.cc | 26 template <class T, class BiasType = T> class DepthwiseConvolutionTestBase : public ::testing::Test {
68 bias = std::vector<BiasType>(output_channels); in DepthwiseConvolutionTestBase()
109 std::vector<BiasType> bias;
|
| /external/armnn/src/armnnTestUtils/ |
| D | CreateWorkload.hpp | 2140 using BiasType = armnn::ResolveType<biasDataType>; in CreatePreCompiledWorkloadTest() typedef
2141 std::vector<BiasType> biasData(biasLength); in CreatePreCompiledWorkloadTest()
2142 std::fill(biasData.begin(), biasData.end(), static_cast<BiasType>(0)); in CreatePreCompiledWorkloadTest()
|