Searched defs:num_y (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | reduction_ops_test.cc | 27 static Graph* ToScalar(const string& reduce, int num_x, int num_y) { in ToScalar()
39 static Graph* ColReduce(const string& reduce, int num_x, int num_y) { in ColReduce()
50 static Graph* RowReduce(const string& reduce, int num_x, int num_y) { in RowReduce()
61 static Graph* ThreeDYReduce(const string& reduce, int num_y, int num_z) { in ThreeDYReduce()
72 static Graph* ThreeDXZReduce(const string& reduce, int num_y, int num_z) { in ThreeDXZReduce()
89 int num_x, int num_y) { in ReduceToScalar()
101 int num_y) { in DoRowReduce()
113 int num_y) { in DoColReduce()
125 int num_y) { in Do3DYReduce()
137 int num_y) { in Do3DXZReduce()
[all …]
|
| D | scan_ops_test.cc | 36 static Graph* ColCumsum(int num_x, int num_y, bool reverse = false) { in ColCumsum()
47 static Graph* RowCumsum(int num_x, int num_y, bool reverse = false) { in RowCumsum()
58 static Graph* ThreeDYCumsum(int num_y, int num_z, bool reverse = false) { in ThreeDYCumsum()
82 const string& device, int num_x, int num_y, in DoRowCumsum()
94 const string& device, int num_x, int num_y, in DoColCumsum()
106 const string& device, int num_x, int num_y, in Do3DYCumsum()
133 const int num_y = state.range(1); in BM_Sum2DRowCumsumGPU() local
141 const int num_y = state.range(1); in BM_Sum2DColumnCumsumGPU() local
149 const int num_y = state.range(1); in BM_Sum3DYCumsumGPU() local
164 const int num_y = state.range(1); in BM_Sum2DRowCumsumGPU_reverse() local
[all …]
|
| /external/mesa3d/src/panfrost/lib/ |
| D | pan_invocation.c | 46 unsigned num_y, in panfrost_pack_work_groups_compute()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | gradients.cc | 104 const int num_y = n->num_outputs(); in AddSymGrad() local
291 const int num_y = y_grad_node_outputs_.size(); in InitBackprop() local
355 const int num_y = n->num_outputs(); in Compute() local
|
| D | function.cc | 1446 const int num_y = static_cast<int>(gbody->ret_nodes.size()); in Compute() local
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_fft.cc | 328 uint64 num_y, fft::Type type, in Create2dPlan()
343 Stream *stream, uint64 num_x, uint64 num_y, fft::Type type, in Create2dPlanWithScratchAllocator()
359 uint64 num_y, uint64 num_z, in Create3dPlan()
376 Stream *stream, uint64 num_x, uint64 num_y, uint64 num_z, fft::Type type, in Create3dPlanWithScratchAllocator()
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_fft.cc | 403 uint64 num_y, fft::Type type, in Create2dPlan()
417 Stream *stream, uint64 num_x, uint64 num_y, fft::Type type, in Create2dPlanWithScratchAllocator()
432 uint64 num_y, uint64 num_z, in Create3dPlan()
447 Stream *stream, uint64 num_x, uint64 num_y, uint64 num_z, fft::Type type, in Create3dPlanWithScratchAllocator()
|
| /external/tensorflow/tensorflow/cc/framework/ |
| D | gradients.cc | 450 const int num_y = n->num_outputs(); in AddGradients() local
|