Searched refs:bias_add_grad (Results 1 – 6 of 6) sorted by relevance
156 bias_add_grad = gradients_impl.gradients(159 output_tensor, np_input.shape, bias_add_grad, bias.shape)182 bias_add_grad = gradients_impl.gradients(186 bias_add_grad,
92 d_b_ru = nn_ops.bias_add_grad(d_r_bar_u_bar)96 d_b_c = nn_ops.bias_add_grad(d_c_bar)
291 b_grad = nn_ops.bias_add_grad(dicfo)
249 csinfo_.bias_add_grad = "BiasAddGrad"; in MklLayoutRewritePass()542 minfo_.push_back({csinfo_.conv2d_grad_filter, csinfo_.bias_add_grad, in MklLayoutRewritePass()683 string bias_add_grad; member1027 if (m->type_string() == csinfo_.bias_add_grad) { in GetConv2DBackpropFilterOrBiasAddGrad()1050 e->dst()->type_string() == csinfo_.bias_add_grad && in GetConv2DBackpropFilterOrBiasAddGrad()2992 CHECK_EQ(((m->type_string() == csinfo_.bias_add_grad && in MergeConv2DBackpropFilterWithBiasAddGrad()2994 ((n->type_string() == csinfo_.bias_add_grad && in MergeConv2DBackpropFilterWithBiasAddGrad()2999 Node* badd = m->type_string() == csinfo_.bias_add_grad ? m : n; in MergeConv2DBackpropFilterWithBiasAddGrad()3000 Node* fltr = m->type_string() == csinfo_.bias_add_grad ? n : m; in MergeConv2DBackpropFilterWithBiasAddGrad()3132 if (((m->type_string() == csinfo_.bias_add_grad && in MergeNode()[all …]
766 gen_nn_ops.bias_add_grad,771 lambda x: gen_nn_ops.bias_add_grad(x, data_format="NCHW"),
341 gen_nn_ops.bias_add_grad(