Searched refs:input_bias_dim (Results 1 – 1 of 1) sorted by relevance
416 DimensionHandle input_bias_dim = c->Dim(input_shape, 1); in BiasAddShape() local418 TF_RETURN_IF_ERROR(c->Merge(input_bias_dim, bias_dim, &merged_bias_dim)); in BiasAddShape()428 DimensionHandle input_bias_dim = c->Dim(input_shape, -1); in BiasAddShape() local430 TF_RETURN_IF_ERROR(c->Merge(input_bias_dim, bias_dim, &merged_bias_dim)); in BiasAddShape()