| /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
109 int32 bias_size) { in BiasGradNHWC_Naive()
121 int32 bias_size, int32 image_size) { in BiasGradNCHW_Naive()
132 T* __restrict__ bias_backprop, int32 bias_size) { in BiasGradNHWC_SharedAtomics()
156 int32 batch, int32 bias_size, int32 image_size, int group_size) { in BiasGradNCHW_SharedAtomics()
219 const int32 bias_size = channel; in compute() local
|
| D | matmul_op_test.cc | 202 const int bias_size = n; in VerifyBiasAddTensorsNear() local
|
| D | conv_ops_test.cc | 738 const int bias_size = filter_count; in VerifyBiasAddTensorsNear() 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 | 250 int bias_size = GetShape(filter_)[0]; local
|
| D | nnapi_delegate_device_selection_test.cc | 715 int bias_size = GetShape(filter_)[0]; local
|
| D | nnapi_delegate_test.cc | 645 int bias_size = GetShape(filter_)[0]; in ConvolutionOpModel() local
1041 const int bias_size = GetShape(filter_)[0]; in PerChannelQuantizedConvolutionWithConstantFilterOpModel() local
1141 int bias_size = GetShape(filter_)[3]; in DepthwiseConvolutionOpModel() local
|
| D | nnapi_delegate.cc | 4122 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 | 599 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 | 42 int bias_size = GetShape(filter_)[0]; in QuantizedTransposeConvOpModel() local
|
| D | conv_test.cc | 43 int bias_size = GetShape(filter_)[0]; in QuantizedConvolutionOpModel() local
|
| /external/tensorflow/tensorflow/lite/delegates/hexagon/builders/ |
| D | conv_2d_helpers.cc | 114 const int bias_size = NumElements(&bias_tensor); in ProcessPerChannelQuantizedBias() local
|
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_fused_ops_test.cc | 131 const int bias_size = filter_count; in VerifyBiasAddTensorsClose() local
149 int filter_size, int filter_count, int bias_size, in VerifyFusedTensorsClose()
350 const int bias_size = filter_count; in VerifyFusedConv2D() local
614 int bias_size, in VerifyFusedDepthwiseConv2D()
|
| /external/tensorflow/tensorflow/lite/tools/optimize/ |
| D | quantization_utils_test.cc | 466 const size_t bias_size = 2; in TEST_F() local
512 const size_t bias_size = 1; in TEST_F() local
|
| D | quantization_utils.cc | 244 const size_t bias_size, in AdjustWeightsForBiasScale()
|
| D | quantize_model.cc | 1866 uint64_t bias_size; in EnsureBiasScaleCompatibility() local
|
| /external/tensorflow/tensorflow/lite/kernels/internal/ |
| D | common.h | 76 inline void BiasAndClamp(float clamp_min, float clamp_max, int bias_size, in BiasAndClamp()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| D | optimize.cc | 1059 int64_t bias_size = bias_and_slice.first; in matchAndRewrite() local
|