Searched refs:bias_dim (Results 1 – 3 of 3) sorted by relevance
77 DimensionHandle bias_dim = c->Dim(bias_shape, 0); in __anon4e6e467d0102() local78 int64 bias_dim_val = c->Value(bias_dim); in __anon4e6e467d0102()
341 bias_dim = -1343 bias_dim = 2 if stacked_value else 1344 bias_shape = [x_shape[bias_dim]]
251 DimensionHandle bias_dim = c->Dim(bias_shape, 0); in BiasAddShape() local272 TF_RETURN_IF_ERROR(c->Merge(input_bias_dim, bias_dim, &merged_bias_dim)); in BiasAddShape()284 TF_RETURN_IF_ERROR(c->Merge(input_bias_dim, bias_dim, &merged_bias_dim)); in BiasAddShape()