Searched refs:bias_index (Results 1 – 4 of 4) sorted by relevance
165 int32 bias_index = blockIdx.x % bias_size; in BiasGradNCHW_SharedAtomics() local174 (batch * bias_size + bias_index) * image_size + image_offset); in BiasGradNCHW_SharedAtomics()194 GpuAtomicAdd(bias_backprop + bias_index, T(data)); in BiasGradNCHW_SharedAtomics()206 GpuAtomicAdd(bias_backprop + bias_index, T(s_data[0])); in BiasGradNCHW_SharedAtomics()
298 virtual inline void GetBiasSizeInMklOrder(size_t bias_index, in GetBiasSizeInMklOrder() argument300 const Tensor& bias = MklGetInput(context_, bias_index); in GetBiasSizeInMklOrder()
688 int bias_index = (flatten_index++ / slice_size) % bias_size; in matchAndRewrite() local690 new_bias_values[bias_index] = in matchAndRewrite()691 new_bias_values[bias_index] + *fp_it * cst_value; in matchAndRewrite()
3106 int bias_index = -1; in Map() local3107 mapping_args.context->AddTensors(mapping_args.context, 1, &bias_index); in Map()3108 TfLiteTensor* bias_tensor = &mapping_args.context->tensors[bias_index]; in Map()