Searched refs:bias_offset (Results 1 – 2 of 2) sorted by relevance
59 int32 bias_offset = index % bias_size; in BiasNHWCKernel() local60 output[index] = ldg(input + index) + ldg(bias + bias_offset); in BiasNHWCKernel()69 int32 bias_offset = index2 % bias_size; in BiasNCHWKernel() local70 output[index] = ldg(input + index) + ldg(bias + bias_offset); in BiasNCHWKernel()105 int32 bias_offset = index % bias_size; in BiasGradNHWC_Naive() local106 CudaAtomicAdd(bias_backprop + bias_offset, ldg(output_backprop + index)); in BiasGradNHWC_Naive()117 int32 bias_offset = index2 % bias_size; in BiasGradNCHW_Naive() local118 CudaAtomicAdd(bias_backprop + bias_offset, ldg(output_backprop + index)); in BiasGradNCHW_Naive()137 int32 bias_offset = index % bias_size; in BiasGradNHWC_SharedAtomics() local138 CudaAtomicAdd(s_data + bias_offset, AccT(ldg(output_backprop + index))); in BiasGradNHWC_SharedAtomics()[all …]
137 Tensor bias_offset(DT_FLOAT, {weights_cols}); in FuseScaleOffsetToConvWeights() local138 auto bias_offset_vector = bias_offset.flat<float>(); in FuseScaleOffsetToConvWeights()160 SetNodeTensorAttr<float>("value", bias_offset, &bias_offset_node); in FuseScaleOffsetToConvWeights()