Home
last modified time | relevance | path

Searched defs:per_channel_scale (Results 1 – 10 of 10) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/internal/
Dtensor_utils.h64 float* __restrict__ result, const float* per_channel_scale, in MatrixBatchVectorMultiplyAccumulate()
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Dsse_tensor_utils.cc100 float* __restrict__ result, const float* per_channel_scale, in SseMatrixBatchVectorMultiplyAccumulateImpl()
273 float* __restrict__ result, const float* per_channel_scale, in SseMatrixBatchVectorMultiplyAccumulate()
Dneon_tensor_utils.cc435 float* __restrict__ result, const float* per_channel_scale, in DotprodMatrixBatchFourVectorMultiplyAccumulate()
564 float* __restrict__ result, const float* per_channel_scale, in DotprodMatrixBatchFourVectorMultiplyAccumulate()
597 float* __restrict__ result, const float* per_channel_scale, in DotprodMatrixBatchPaddedFourVectorMultiplyAccumulate()
1246 int n_batch, float* __restrict__ result, const float* per_channel_scale, in NeonMatrixBatchVectorMultiplyAccumulateImpl()
1374 int n_batch, float* __restrict__ result, const float* per_channel_scale, in NeonMatrixBatchVectorMultiplyAccumulate()
Dsse_tensor_utils.h56 int n_batch, float* __restrict__ result, const float* per_channel_scale, in MatrixBatchVectorMultiplyAccumulate()
Dneon_tensor_utils.h58 int n_batch, float* __restrict__ result, const float* per_channel_scale, in MatrixBatchVectorMultiplyAccumulate()
Doptimized_ops.h1432 const float* per_channel_scale, int32_t* input_offset, in HybridConvPerChannel()
/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dconv.h194 const float* per_channel_scale, int32_t* input_offset) { in HybridConvPerChannel()
Dportable_tensor_utils.h78 int n_batch, float* __restrict__ result, const float* per_channel_scale, in MatrixBatchVectorMultiplyAccumulate()
Dportable_tensor_utils.cc166 int n_batch, float* __restrict__ result, const float* per_channel_scale, in PortableMatrixBatchVectorMultiplyAccumulate()
/external/tensorflow/tensorflow/lite/kernels/internal/reference/integer_ops/
Ddepthwise_conv.h214 const float* per_channel_scale, int32_t* input_offset) { in DepthwiseConvHybridPerChannel()