| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| bias_ops.cc | 42 const TensorShape bias_shape = ctx->InputShape(1); variable
47 OP_REQUIRES(ctx, TensorShapeUtils::IsVector(bias_shape),
49 bias_shape.DebugString()));
57 ctx, bias_shape.dim_size(0) == input_shape.dim_size(feature_dim),
61 bias_shape.DebugString(), " vs. ", input_shape.DebugString()));
|
| /external/tensorflow/tensorflow/contrib/fused_conv/ops/ |
| fused_conv2d_bias_activation_op.cc | 74 ShapeHandle bias_shape;
76 TF_RETURN_IF_ERROR(c->WithRank(c->input(2), 1, &bias_shape));
77 DimensionHandle bias_dim = c->Dim(bias_shape, 0);
|
| /external/tensorflow/tensorflow/contrib/lite/toco/graph_transformations/ |
| fuse_binary_into_preceding_affine.cc | 44 const Shape& bias_shape = bias.shape(); local
52 const int depth = bias_shape.dims(bias_shape.dimensions_count() - 1);
101 const Shape& bias_shape = bias.shape(); local
115 bias_shape.dims(bias_shape.dimensions_count() - 1)) {
|
| fuse_binary_into_following_affine.cc | 71 const Shape& bias_shape = bias.shape(); local
81 CHECK_EQ(output_depth, bias_shape.dims(bias_shape.dimensions_count() - 1));
|
| propagate_fixed_sizes.cc | 805 const auto& bias_shape = bias_array.shape(); local
806 CHECK_GE(bias_shape.dimensions_count(), 1);
818 CHECK_EQ(fc_output_depth, bias_shape.dims(0));
[all...] |
| /external/tensorflow/tensorflow/contrib/kernel_methods/python/mappers/ |
| random_fourier_features.py | 149 bias_shape = [self._output_dim]
157 low=0.0, high=2 * np.pi, size=bias_shape),
|
| /external/tensorflow/tensorflow/contrib/fused_conv/python/ops/ |
| fused_conv2d_bias_activation_benchmark.py | 58 bias_shape = [filter_shape[-1]]
59 bias = variables.Variable(random_ops.truncated_normal(bias_shape))
103 bias_shape = [filter_shape[-1]]
104 bias = variables.Variable(random_ops.truncated_normal(bias_shape))
|
| /external/tensorflow/tensorflow/cc/gradients/ |
| nn_grad_test.cc | 164 TensorShape bias_shape({5});
166 auto bias = Placeholder(scope_, DT_FLOAT, Placeholder::Shape(bias_shape));
168 RunTest({x, bias}, {shape, bias_shape}, {y}, {shape});
|
| /external/tensorflow/tensorflow/python/keras/_impl/keras/ |
| backend.py | [all...] |
| /external/tensorflow/tensorflow/python/ops/ |
| nn_grad.py | 312 bias_shape = array_ops.shape(received_grad)
316 array_ops.ones_like(shape[:-3]), bias_shape,
322 [array_ops.ones_like(shape[:-1]), bias_shape], 0)
[all...] |
| /external/tensorflow/tensorflow/core/framework/ |
| common_shape_fns.cc | 215 ShapeHandle bias_shape; local
216 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 1, &bias_shape));
217 DimensionHandle bias_dim = c->Dim(bias_shape, 0);
229 // Merge the length of bias_shape into the third to last dimension
[all...] |
