Searched defs:per_channel_scale (Results 1 – 10 of 10) sorted by relevance
| /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 | sse_tensor_utils.h | 61 int n_batch, float* __restrict__ result, const float* per_channel_scale, in MatrixBatchVectorMultiplyAccumulate()
|
| D | neon_tensor_utils.h | 58 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()
|