Searched defs:bias_offset (Results 1 – 4 of 4) sorted by relevance
58 int32 bias_offset = index % bias_size; in BiasNHWCKernel() local70 int32 bias_offset = index2 % bias_size; in BiasNCHWKernel() local111 int32 bias_offset = index % bias_size; in BiasGradNHWC_Naive() local124 int32 bias_offset = index2 % bias_size; in BiasGradNCHW_Naive() local143 int32 bias_offset = index % bias_size; in BiasGradNHWC_SharedAtomics() local183 int bias_offset = threadIdx.x % 32; in BiasGradNCHW_SharedAtomics() local
45 const int32_t bias_offset = h * w * params.input_zero_point * params.kernel_zero_point; in TEST() local95 const int32_t bias_offset = h * w * params.input_zero_point * params.kernel_zero_point; in TEST() local156 const int32_t bias_offset = h * w * params.input_zero_point * params.kernel_zero_point; in TEST() local213 const int32_t bias_offset = h * w * params.input_zero_point * params.kernel_zero_point; in TEST() local278 const int32_t bias_offset = h * w * params.input_zero_point * params.kernel_zero_point; in TEST() local326 const int32_t bias_offset = h * w * params.input_zero_point * params.kernel_zero_point; in TEST() local387 const int32_t bias_offset = h * w * params.input_zero_point * params.kernel_zero_point; in TEST() local438 const int32_t bias_offset = h * w * params.input_zero_point * params.kernel_zero_point; in TEST() local
136 Tensor bias_offset(DT_FLOAT, {weights_cols}); in FuseScaleOffsetToConvWeights() local
129 size_t bias_offset = 0; in AssignLayerWeightsBiases() local