| /external/tensorflow/tensorflow/cc/framework/ |
| gradient_checker_test.cc | 127 TensorShape y_shape({3, 2});
131 auto y = Const(scope, {1.0, 2.0, 3.0, 4.0, 5.0, 6.0}, y_shape);
147 TensorShape y_shape = TensorShape({5, 1}); local
150 scope, {x}, {x_shape}, y.output, {y_shape, y_shape}, &max_error)));
162 TensorShape y_shape({2, 1, 2, 3});
165 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});
114 TensorShape y_shape({2, 5, 3})
[all...] |
| math_grad_test.cc | 581 TensorShape y_shape = shapes[1]; local
586 Placeholder(root_, DataTypeToEnum<T>::v(), Placeholder::Shape(y_shape));
596 root_, {x, y}, {x_shape, y_shape}, {z}, {z_shape}, &max_error)));
619 TensorShape y_shape; local
622 y_shape = ty ? TensorShape({b, n, k}) : TensorShape({b, k, n});
625 y_shape = ty ? TensorShape({n, k}) : TensorShape({k, n});
627 shapes->push_back(y_shape);
723 const TensorShape& y_shape) {
727 scope_, x, x_init_value, y, y_shape, &max_error)));
739 TensorShape y_shape({2, 5})
[all...] |
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| xla_builder_test.cc | 192 const auto& y_shape = ShapeUtil::MakeShape(S32, {2, 4}); local
194 auto y = Parameter(&b, 1, y_shape, "y");
[all...] |
| /external/tensorflow/tensorflow/core/common_runtime/ |
| function_test.cc | 1548 auto y_shape = ops::DynamicStitch( local
[all...] |
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| constant_folding.cc | 2663 const TensorShapeProto& y_shape = local
[all...] |
