Searched defs:bias_offset (Results 1 – 2 of 2) sorted by relevance
59 int32 bias_offset = index % bias_size; in BiasNHWCKernel() local69 int32 bias_offset = index2 % bias_size; in BiasNCHWKernel() local105 int32 bias_offset = index % bias_size; in BiasGradNHWC_Naive() local117 int32 bias_offset = index2 % bias_size; in BiasGradNCHW_Naive() local137 int32 bias_offset = index % bias_size; in BiasGradNHWC_SharedAtomics() local178 int bias_offset = threadIdx.x % 32; in BiasGradNCHW_SharedAtomics() local
137 Tensor bias_offset(DT_FLOAT, {weights_cols}); in FuseScaleOffsetToConvWeights() local