| /external/tensorflow/tensorflow/cc/framework/ |
| gradient_checker_test.cc | 111 TensorShape y_shape({3, 2});
115 auto y = Const(scope, {1.0, 2.0, 3.0, 4.0, 5.0, 6.0}, y_shape);
131 TensorShape y_shape = TensorShape({5, 1}); local
134 scope, {x}, {x_shape}, y.output, {y_shape, y_shape}, &max_error)));
146 TensorShape y_shape({2, 1, 2, 3});
149 scope, xs, {x_shape, x_shape}, {y}, {y_shape}, &max_error)));
|
| /external/tensorflow/tensorflow/cc/gradients/ |
| array_grad_test.cc | 36 const TensorShape& y_shape) {
40 scope_, {x}, {x_shape}, {y}, {y_shape}, &max_error)));
62 TensorShape y_shape({2, 1, 2, 3});
63 RunTest(xs, {x_shape, x_shape}, {y}, {y_shape});
72 TensorShape y_shape({1, 2, 2, 3});
73 RunTest(xs, {x_shape, x_shape}, {y}, {y_shape});
107 TensorShape y_shape = TensorShape({5, 1}); local
108 RunTest({x}, {x_shape}, y.output, {y_shape, y_shape});
115 TensorShape y_shape({5, 2, 5, 2})
[all...] |
| math_grad_test.cc | 583 TensorShape y_shape = shapes[1]; local
588 Placeholder(root_, DataTypeToEnum<T>::v(), Placeholder::Shape(y_shape));
598 root_, {x, y}, {x_shape, y_shape}, {z}, {z_shape}, &max_error)));
621 TensorShape y_shape; local
624 y_shape = ty ? TensorShape({b, n, k}) : TensorShape({b, k, n});
627 y_shape = ty ? TensorShape({n, k}) : TensorShape({k, n});
629 shapes->push_back(y_shape);
725 const TensorShape& y_shape) {
729 scope_, x, x_init_value, y, y_shape, &max_error)));
741 TensorShape y_shape({2, 5})
[all...] |
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| arithmetic_optimizer.cc | 262 PartialTensorShape y_shape = GetInputShape(input_y, node_map); local
263 if (x_shape.unknown_rank() || y_shape.unknown_rank() ||
264 x_shape.dims() != y_shape.dims()) {
268 if (x_shape.dim_size(i) == -1 || y_shape.dim_size(i) == -1 ||
269 x_shape.dim_size(i) != y_shape.dim_size(i)) {
[all...] |
| constant_folding.cc | 1491 const TensorShapeProto& y_shape = local
[all...] |
| /external/tensorflow/tensorflow/core/common_runtime/ |
| function_test.cc | 1204 auto y_shape = ops::DynamicStitch( local
[all...] |
