Searched defs:bias_index (Results 1 – 5 of 5) sorted by relevance
303 virtual inline void GetBiasSizeInMklOrder(size_t bias_index, in GetBiasSizeInMklOrder()
168 int32 bias_index = blockIdx.x % bias_size; in BiasGradNCHW_SharedAtomics() local
995 Operation *op, int bias_index, const std::vector<int> &input_indices, in ShouldCheckBiasScale()1030 Operation *op, int bias_index, const std::vector<int> &input_indices, in SetBiasParamsWithAdjustments()
1084 int bias_index = (flatten_index++ / slice_size) % bias_size; in matchAndRewrite() local
3570 int bias_index = -1; in Map() local