Searched refs:y_flat (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/cc/tutorials/ |
| D | example_trainer.cc | 90 auto y_flat = y.flat<float>(); in DebugString() local
94 Eigen::Tensor<float, 0, Eigen::RowMajor> lambda = (x_flat * y_flat).sum(); in DebugString()
96 lambda(), x_flat(0), x_flat(1), y_flat(0), y_flat(1)); in DebugString()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | rpcbench_test.cc | 161 auto y_flat = y.flat<float>(); in DebugString() local
165 x_flat(1), y_flat(0), y_flat(1)); in DebugString()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | dot_operation_test.cc | 559 auto y_flat = Reshape(y, {0, 1, 2, 3}, {4, 2, 2}); in XLA_TYPED_TEST() local
567 auto y_slice = Slice(y_flat, {i, 0, 0}, {i + 1, 2, 2}, {1, 1, 1}); in XLA_TYPED_TEST()
|