| /external/tensorflow/tensorflow/core/kernels/ |
| reduction_ops_test.cc | 27 static Graph* ToScalar(const string& reduce, int num_x, int num_y) {
29 Tensor data(DataTypeToEnum<T>::value, TensorShape({num_x, num_y}));
39 static Graph* ColReduce(const string& reduce, int num_x, int num_y) {
41 Tensor data(DT_FLOAT, TensorShape({num_x, num_y}));
50 static Graph* RowReduce(const string& reduce, int num_x, int num_y) {
52 Tensor data(DT_FLOAT, TensorShape({num_x, num_y}));
61 static Graph* ThreeDYReduce(const string& reduce, int num_y, int num_z) {
63 Tensor data(DT_FLOAT, TensorShape({4, num_y, num_z}));
72 static Graph* ThreeDXZReduce(const string& reduce, int num_y, int num_z) {
74 Tensor data(DT_FLOAT, TensorShape({4, num_y, num_z}))
89 testing::ItemsProcessed(static_cast<int64>(iters) * num_x * num_y); local
97 testing::ItemsProcessed(static_cast<int64>(iters) * num_x * num_y); local
105 testing::ItemsProcessed(static_cast<int64>(iters) * num_x * num_y); local
113 testing::ItemsProcessed(static_cast<int64>(iters) * num_x * num_y); local
121 testing::ItemsProcessed(static_cast<int64>(iters) * num_x * num_y); local
[all...] |
| /external/tensorflow/tensorflow/cc/framework/ |
| gradients.cc | 449 const int num_y = n->num_outputs(); local
451 dy.resize(num_y, {nullptr, 0});
453 for (int i = 0; i < num_y; ++i) {
492 if (IsPrimitiveOpWithNoGrad(n->type_string()) || num_no_grad == num_y) {
503 if (num_no_grad > 0 && num_no_grad < num_y) {
|
| /external/tensorflow/tensorflow/core/graph/ |
| gradients.cc | 82 const int num_y = n->num_outputs(); local
83 CHECK_EQ(num_y, grads.size());
89 // The gradient node should have num_x + num_y inputs.
104 CHECK_EQ(ndef.input_size(), num_x + num_y);
254 const int num_y = y_grad_node_outputs_.size(); local
255 for (int i = 0; i < num_y; ++i) {
316 // "n" has num_x inputs and num_y outputs.
318 const int num_y = n->num_outputs(); local
324 CHECK_EQ(1, num_y);
330 dy.resize(num_y, {nullptr, 0})
[all...] |
| /external/tensorflow/tensorflow/core/common_runtime/ |
| function.cc | 1504 const int num_y = static_cast<int>(gbody_->ret_nodes.size()); local
[all...] |
