Searched refs:bias_add_grad (Results 1 – 5 of 5) sorted by relevance
198 bias_add_grad = gradients_impl.gradients(201 output_tensor, output_shape, bias_add_grad, bias_shape)
251 csinfo_.bias_add_grad = "BiasAddGrad"; in MklLayoutRewritePass()719 minfo_.push_back({csinfo_.conv2d_grad_filter, csinfo_.bias_add_grad, in MklLayoutRewritePass()912 string bias_add_grad; member1302 if (m->type_string() == csinfo_.bias_add_grad) { in GetConv2DBackpropFilterOrBiasAddGrad()1327 e->dst()->type_string() == csinfo_.bias_add_grad && in GetConv2DBackpropFilterOrBiasAddGrad()3311 CHECK_EQ(((m->type_string() == csinfo_.bias_add_grad && in MergeConv2DBackpropFilterWithBiasAddGrad()3313 ((n->type_string() == csinfo_.bias_add_grad && in MergeConv2DBackpropFilterWithBiasAddGrad()3318 Node* badd = m->type_string() == csinfo_.bias_add_grad ? m : n; in MergeConv2DBackpropFilterWithBiasAddGrad()3319 Node* fltr = m->type_string() == csinfo_.bias_add_grad ? n : m; in MergeConv2DBackpropFilterWithBiasAddGrad()3463 if (((m->type_string() == csinfo_.bias_add_grad && in MergeNode()[all …]
880 gen_nn_ops.bias_add_grad,885 lambda x: gen_nn_ops.bias_add_grad(x, data_format="NCHW"),
351 gen_nn_ops.bias_add_grad(
1576 db = gen_nn_ops.bias_add_grad(dx, 'NHWC')