Searched defs:bias_offset (Results 1 – 3 of 3) sorted by relevance
58 int32 bias_offset = index % bias_size; in BiasNHWCKernel() local70 int32 bias_offset = index2 % bias_size; in BiasNCHWKernel() local108 int32 bias_offset = index % bias_size; in BiasGradNHWC_Naive() local121 int32 bias_offset = index2 % bias_size; in BiasGradNCHW_Naive() local140 int32 bias_offset = index % bias_size; in BiasGradNHWC_SharedAtomics() local180 int bias_offset = threadIdx.x % 32; in BiasGradNCHW_SharedAtomics() local
136 Tensor bias_offset(DT_FLOAT, {weights_cols}); in FuseScaleOffsetToConvWeights() local
117 size_t bias_offset = 0; in AssignLayerWeightsBiases() local