Searched refs:x_flat (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/cc/tutorials/ |
| D | example_trainer.cc | 89 auto x_flat = x.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()
119 auto x_flat = x.flat<float>(); in ConcurrentSteps() local
120 x_flat.setRandom(); in ConcurrentSteps()
122 x_flat.square().sum().sqrt().inverse(); in ConcurrentSteps()
123 x_flat = x_flat * inv_norm(); in ConcurrentSteps()
|
| /external/tensorflow/tensorflow/cc/profiler/ |
| D | profiler_test.cc | 107 auto x_flat = x.flat<float>(); in TEST_F() local
108 x_flat.setRandom(); in TEST_F()
110 x_flat.square().sum().sqrt().inverse(); in TEST_F()
111 x_flat = x_flat * inv_norm(); in TEST_F()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | rpcbench_test.cc | 160 auto x_flat = x.flat<float>(); in DebugString() local
164 return strings::Printf("x = [%8.6f %8.6f] y = [%8.6f %8.6f]", x_flat(0), in DebugString()
165 x_flat(1), y_flat(0), y_flat(1)); in DebugString()
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | segment_reduction_ops_test.py | 61 x_flat = x.reshape((indices.size,) + slice_shape)
65 output[index][j] = op1([output[index][j], x_flat[i][j]])
67 output[index] = op1(output[index], x_flat[i])
69 output[index] = x_flat[i]
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | dot_operation_test.cc | 602 auto x_flat = Reshape(x, {0, 1, 2, 3}, {4, 2, 2}); in XLA_TYPED_TEST() local
609 auto x_slice = Slice(x_flat, {i, 0, 0}, {i + 1, 2, 2}, {1, 1, 1}); in XLA_TYPED_TEST()
|