Searched defs:per_channel_scale (Results 1 – 12 of 12) sorted by relevance
| /external/pytorch/torch/testing/_internal/ |
| D | common_quantized.py | 196 def _fake_quantize_per_channel_affine_reference(X, per_channel_scale, per_channel_zero_point, axis,… argument
210 def _fake_quantize_per_channel_affine_grad_reference(dY, X, per_channel_scale, per_channel_zero_poi… argument
|
| /external/tensorflow/tensorflow/lite/kernels/internal/ |
| D | tensor_utils.h | 64 float* __restrict__ result, const float* per_channel_scale, in MatrixBatchVectorMultiplyAccumulate()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | sse_tensor_utils.cc | 203 float* __restrict__ result, const float* per_channel_scale, in Avx2MatrixBatchVectorMultiplyAccumulateImpl()
295 float* __restrict__ result, const float* per_channel_scale, in SseMatrixBatchVectorMultiplyAccumulateImpl()
476 float* __restrict__ result, const float* per_channel_scale, in SseMatrixBatchVectorMultiplyAccumulate()
|
| D | neon_tensor_utils.cc | 460 float* __restrict__ result, const float* per_channel_scale, in DotprodMatrixBatchFourVectorMultiplyAccumulate()
590 float* __restrict__ result, const float* per_channel_scale, in DotprodMatrixBatchFourVectorMultiplyAccumulate()
623 float* __restrict__ result, const float* per_channel_scale, in DotprodMatrixBatchPaddedFourVectorMultiplyAccumulate()
1272 int n_batch, float* __restrict__ result, const float* per_channel_scale, in NeonMatrixBatchVectorMultiplyAccumulateImpl()
1400 int n_batch, float* __restrict__ result, const float* per_channel_scale, in NeonMatrixBatchVectorMultiplyAccumulate()
|
| D | neon_tensor_utils.h | 58 int n_batch, float* __restrict__ result, const float* per_channel_scale, in MatrixBatchVectorMultiplyAccumulate()
|
| D | sse_tensor_utils.h | 61 int n_batch, float* __restrict__ result, const float* per_channel_scale, in MatrixBatchVectorMultiplyAccumulate()
|
| D | optimized_ops.h | 1101 const float* per_channel_scale, int32_t* input_offset, in HybridConvPerChannel()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/integer_ops/ |
| D | depthwise_conv.h | 216 const float* per_channel_scale, int32_t* input_offset) { in DepthwiseConvHybridPerChannel()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | conv.h | 210 const float* per_channel_scale, int32_t* input_offset) { in HybridConvPerChannel()
|
| D | portable_tensor_utils.h | 78 int n_batch, float* __restrict__ result, const float* per_channel_scale, in MatrixBatchVectorMultiplyAccumulate()
|
| D | portable_tensor_utils.cc | 166 int n_batch, float* __restrict__ result, const float* per_channel_scale, in PortableMatrixBatchVectorMultiplyAccumulate()
|
| /external/pytorch/test/quantization/core/ |
| D | test_workflow_ops.py | 98 dY, X, per_channel_scale, per_channel_zero_point, axis, quant_min, quant_max, device): argument
|