Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/contrib/fused_conv/ops/
Dfused_conv2d_bias_activation_op.cc77 DimensionHandle bias_dim = c->Dim(bias_shape, 0); in __anon4e6e467d0102() local
78 int64 bias_dim_val = c->Value(bias_dim); in __anon4e6e467d0102()
/external/tensorflow/tensorflow/python/ops/parallel_for/
Dmath_test.py341 bias_dim = -1
343 bias_dim = 2 if stacked_value else 1
344 bias_shape = [x_shape[bias_dim]]
/external/tensorflow/tensorflow/core/framework/
Dcommon_shape_fns.cc251 DimensionHandle bias_dim = c->Dim(bias_shape, 0); in BiasAddShape() local
272 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()