| /external/tensorflow/tensorflow/core/kernels/ |
| D | bias_op_gpu.cu.cc | 56 T* __restrict__ output, int32 bias_size) { in BiasNHWCKernel()
66 T* __restrict__ output, int32 bias_size, in BiasNCHWKernel()
81 const int32 bias_size = channel; in compute() local
106 int32 bias_size) { in BiasGradNHWC_Naive()
118 int32 bias_size, int32 image_size) { in BiasGradNCHW_Naive()
129 T* __restrict__ bias_backprop, int32 bias_size) { in BiasGradNHWC_SharedAtomics()
153 int32 batch, int32 bias_size, int32 image_size, int group_size) { in BiasGradNCHW_SharedAtomics()
216 const int32 bias_size = channel; in compute() local
|
| D | bias_op.h | 34 const Eigen::Index bias_size = bias.dimension(0); in operator() local
41 const int bias_size = bias.dimension(0); in operator() local
|
| D | matmul_op_test.cc | 202 const int bias_size = n; in VerifyBiasAddTensorsNear() local
|
| D | conv_ops_test.cc | 795 const int bias_size = filter_count; in VerifyBiasAddTensorsNear() local
|
| /external/tensorflow/tensorflow/core/kernels/neon/ |
| D | neon_depthwise_conv_op.cc | 130 int64 bias_size = bias_neon_dims.sizes[0]; in Compute() local
|
| /external/tensorflow/tensorflow/lite/delegates/nnapi/ |
| D | quant_lstm_sup.cc | 130 void DecomposeBiasTensor(const int32_t* biases, int bias_size, in DecomposeBiasTensor()
|
| D | nnapi_delegate_signed_quantization_test.cc | 246 int bias_size = GetShape(filter_)[0]; local
|
| D | nnapi_delegate_device_selection_test.cc | 716 int bias_size = GetShape(filter_)[0]; local
|
| D | nnapi_delegate_test.cc | 601 int bias_size = GetShape(filter_)[0]; in ConvolutionOpModel() local
964 const int bias_size = GetShape(filter_)[0]; in PerChannelQuantizedConvolutionWithConstantFilterOpModel() local
1064 int bias_size = GetShape(filter_)[3]; in DepthwiseConvolutionOpModel() local
|
| D | nnapi_delegate.cc | 3293 const auto bias_size = output_dims->data[1]; in Map() local
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | depthwise_conv_hybrid_test.cc | 62 int bias_size = GetShape(filter_)[3]; in BaseDepthwiseConvolutionOpModel() local
183 const int bias_size = c; in RandomTest() local
|
| D | transpose_conv_test.cc | 590 int bias_size = GetShape(filter_)[0]; in BaseTransposeConvBiasOpModel() local
|
| D | depthwise_conv_test.cc | 63 int bias_size = GetShape(filter_)[3]; in BaseDepthwiseConvolutionOpModel() local
652 const int bias_size = 2400; in TEST_F() local
|
| D | conv_test.cc | 69 int bias_size = GetShape(filter_)[0]; local
|
| /external/tensorflow/tensorflow/lite/delegates/hexagon/builders/tests/ |
| D | transpose_conv_test.cc | 40 int bias_size = GetShape(filter_)[0]; in QuantizedTransposeConvOpModel() local
|
| D | conv_test.cc | 41 int bias_size = GetShape(filter_)[0]; in QuantizedConvolutionOpModel() local
|
| /external/tensorflow/tensorflow/lite/delegates/hexagon/builders/ |
| D | conv_2d_helpers.cc | 98 const int bias_size = NumElements(&bias_tensor); in ProcessPerChannelQuantizedBias() local
|
| /external/tensorflow/tensorflow/lite/tools/optimize/ |
| D | quantization_utils_test.cc | 396 const size_t bias_size = 2; in TEST_F() local
442 const size_t bias_size = 1; in TEST_F() local
|
| D | quantization_utils.cc | 234 const size_t bias_size, in AdjustWeightsForBiasScale()
|
| D | quantize_model.cc | 1392 uint64_t bias_size; in EnsureBiasScaleCompatibility() local
|
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_fused_ops_test.cc | 129 const int bias_size = filter_count; in VerifyBiasAddTensorsClose() local
147 int filter_size, int filter_count, int bias_size, in VerifyFusedTensorsClose()
370 const int bias_size = filter_count; in VerifyFusedConv2D() local
666 int bias_size, in VerifyFusedDepthwiseConv2D()
|
| /external/tensorflow/tensorflow/lite/micro/kernels/ |
| D | depthwise_conv_test.cc | 647 const int bias_size = 4; in TF_LITE_MICRO_TEST() local
790 const int bias_size = 4; in TF_LITE_MICRO_TEST() local
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| D | optimize.cc | 664 int64_t bias_size = bias_and_slice.first; in matchAndRewrite() local
|
| /external/tensorflow/tensorflow/lite/kernels/internal/ |
| D | common.h | 75 inline void BiasAndClamp(float clamp_min, float clamp_max, int bias_size, in BiasAndClamp()
|
| /external/XNNPACK/bench/ |
| D | convolution.cc | 731 const float* kernel, size_t kernel_size, const float* bias, size_t bias_size, in compare_with_convolution_f32_reference_output()
|