Searched refs:input_bias_dim (Results 1 – 1 of 1) sorted by relevance
270 DimensionHandle input_bias_dim = c->Dim(input_shape, 1); in BiasAddShape() local272 TF_RETURN_IF_ERROR(c->Merge(input_bias_dim, bias_dim, &merged_bias_dim)); in BiasAddShape()282 DimensionHandle input_bias_dim = c->Dim(input_shape, -1); in BiasAddShape() local284 TF_RETURN_IF_ERROR(c->Merge(input_bias_dim, bias_dim, &merged_bias_dim)); in BiasAddShape()