Searched defs:bias_index (Results 1 – 8 of 8) sorted by relevance
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()
15 constexpr int64_t bias_index = 1; variable
61 unsigned bias_index; in patch_conv2d() local
303 virtual inline void GetBiasSizeInMklOrder(size_t bias_index, in GetBiasSizeInMklOrder()
168 int32 bias_index = blockIdx.x % bias_size; in BiasGradNCHW_SharedAtomics() local
395 int bias_index = nir_tex_instr_src_index(tex, nir_tex_src_bias); in evalute_active_lod() local
1084 int bias_index = (flatten_index++ / slice_size) % bias_size; in matchAndRewrite() local
3570 int bias_index = -1; in Map() local