| /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}));
88 const string& reduce, int num_x, int num_y) {
89 testing::ItemsProcessed(static_cast<int64>(iters) * num_x * num_y);
90 testing::BytesProcessed(static_cast<int64>(iters) * num_x * num_y *
92 test::Benchmark(device, ToScalar<T>(reduce, num_x, num_y)).Run(iters)
[all...] |
| /external/tensorflow/tensorflow/stream_executor/ |
| fft.h | 95 virtual std::unique_ptr<Plan> Create1dPlan(Stream *stream, uint64 num_x,
99 virtual std::unique_ptr<Plan> Create2dPlan(Stream *stream, uint64 num_x,
104 virtual std::unique_ptr<Plan> Create3dPlan(Stream *stream, uint64 num_x,
110 Stream *stream, uint64 num_x, Type type, bool in_place_fft,
115 Stream *stream, uint64 num_x, uint64 num_y, Type type, bool in_place_fft,
120 Stream *stream, uint64 num_x, uint64 num_y, uint64 num_z, Type type,
215 std::unique_ptr<fft::Plan> Create1dPlan(Stream *stream, uint64 num_x, \
218 std::unique_ptr<fft::Plan> Create2dPlan(Stream *stream, uint64 num_x, \
222 Stream *stream, uint64 num_x, uint64 num_y, uint64 num_z, \
225 Stream *stream, uint64 num_x, fft::Type type, bool in_place_fft,
[all...] |
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| cuda_fft.cc | 326 std::unique_ptr<fft::Plan> CUDAFft::Create1dPlan(Stream *stream, uint64 num_x,
330 uint64 elem_count[1] = {num_x};
343 Stream *stream, uint64 num_x, fft::Type type, bool in_place_fft,
346 uint64 elem_count[1] = {num_x};
357 std::unique_ptr<fft::Plan> CUDAFft::Create2dPlan(Stream *stream, uint64 num_x,
361 uint64 elem_count[2] = {num_x, num_y};
372 Stream *stream, uint64 num_x, uint64 num_y, fft::Type type,
375 uint64 elem_count[2] = {num_x, num_y};
386 std::unique_ptr<fft::Plan> CUDAFft::Create3dPlan(Stream *stream, uint64 num_x,
391 uint64 elem_count[3] = {num_x, num_y, num_z}
[all...] |
| /external/tensorflow/tensorflow/core/graph/ |
| gradients.cc | 81 const int num_x = n->num_inputs(); local
89 // The gradient node should have num_x + num_y inputs.
90 std::vector<NodeOut> n_inputs(num_x);
104 CHECK_EQ(ndef.input_size(), num_x + num_y);
316 // "n" has num_x inputs and num_y outputs.
317 const int num_x = n->num_inputs(); local
344 // Adds a gradient node with num_x + num_y inputs and num_x
353 graph_->AddEdge(dy[i].node, dy[i].index, grad, num_x + i);
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| function.cc | 1527 const size_t num_x = fbody_->arg_nodes.size(); local
[all...] |
