Home
last modified time | relevance | path

Searched refs:input_bias_dim (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/core/framework/
Dcommon_shape_fns.cc270 DimensionHandle input_bias_dim = c->Dim(input_shape, 1); in BiasAddShape() local
272 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() local
284 TF_RETURN_IF_ERROR(c->Merge(input_bias_dim, bias_dim, &merged_bias_dim)); in BiasAddShape()