Searched refs:y_flat (Results 1 - 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/cc/tutorials/ |
| H A D | example_trainer.cc | 89 auto y_flat = y.flat<float>(); local
93 Eigen::Tensor<float, 0, Eigen::RowMajor> lambda = (x_flat * y_flat).sum();
95 lambda(), x_flat(0), x_flat(1), y_flat(0), y_flat(1));
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| H A D | rpcbench_test.cc | 161 auto y_flat = y.flat<float>(); local
165 x_flat(1), y_flat(0), y_flat(1));
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| H A D | dot_operation_test.cc | 492 auto y_flat = builder.Reshape(y, {0, 1, 2, 3}, {4, 2, 2}); local
500 auto y_slice = builder.Slice(y_flat, {i, 0, 0}, {i + 1, 2, 2}, {1, 1, 1});
|
Completed in 235 milliseconds