Searched defs:bias_size (Results 1 – 10 of 10) sorted by relevance
34 const int64_t bias_size = bias.dimension(0); in operator() local41 const int bias_size = bias.dimension(0); in operator() local
57 T* output, int32 bias_size) { in BiasNHWCKernel()66 T* output, int32 bias_size, int32 image_size) { in BiasNCHWKernel()80 const int32 bias_size = channel; in compute() local103 T* bias_backprop, int32 bias_size) { in BiasGradNHWC_Naive()113 T* bias_backprop, int32 bias_size, in BiasGradNCHW_Naive()127 T* bias_backprop, int32 bias_size) { in BiasGradNHWC_SharedAtomics()150 int32 bias_size, int32 image_size, in BiasGradNCHW_SharedAtomics()201 const int32 bias_size = channel; in compute() local
105 const int bias_size = filter_count; in VerifyBiasAddTensorsClose() local
759 const int bias_size = filter_count; in VerifyBiasAddTensorsNear() local
130 int64 bias_size = bias_neon_dims.sizes[0]; in Compute() local
52 int bias_size = GetShape(filter_)[3]; in BaseDepthwiseConvolutionOpModel() local
54 int bias_size = GetShape(filter_)[0]; in BaseConvolutionOpModel() local
275 int bias_size = GetShape(filter_)[0]; in ConvolutionOpModel() local454 int bias_size = GetShape(filter_)[3]; in DepthwiseConvolutionOpModel() local
195 const int bias_size = bias_shape.FlatSize(); in AddBiasAndEvalActivationFunction() local
290 const int bias_size = bias_shape.FlatSize(); in AddBiasAndEvalActivationFunction() local