Searched defs:num_y (Results 1 – 7 of 7) 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() 88 const string& reduce, int num_x, int num_y) { in ReduceToScalar() 96 int num_x, int num_y) { in DoRowReduce() 104 int num_x, int num_y) { in DoColReduce() 112 int num_x, int num_y) { in Do3DYReduce() 120 int num_x, 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() 77 static void DoRowCumsum(int iters, const string& device, int num_x, int num_y, in DoRowCumsum() 85 static void DoColCumsum(int iters, const string& device, int num_x, int num_y, in DoColCumsum() 93 static void Do3DYCumsum(int iters, const string& device, int num_x, int num_y, in Do3DYCumsum() 111 static void BM_Sum2DRowCumsumGPU(int iters, int num_x, int num_y) { in BM_Sum2DRowCumsumGPU() 116 static void BM_Sum2DColumnCumsumGPU(int iters, int num_x, int num_y) { in BM_Sum2DColumnCumsumGPU() 121 static void BM_Sum3DYCumsumGPU(int iters, int num_x, int num_y) { in BM_Sum3DYCumsumGPU() 131 static void BM_Sum2DRowCumsumGPU_reverse(int iters, int num_x, int num_y) { in BM_Sum2DRowCumsumGPU_reverse() [all …]
|
/external/tensorflow/tensorflow/core/graph/ |
D | gradients.cc | 103 const int num_y = n->num_outputs(); in AddSymGrad() local 282 const int num_y = y_grad_node_outputs_.size(); in InitBackprop() local 346 const int num_y = n->num_outputs(); in Compute() local
|
/external/tensorflow/tensorflow/stream_executor/cuda/ |
D | cuda_fft.cc | 325 uint64 num_y, fft::Type type, in Create2dPlan() 340 Stream *stream, uint64 num_x, uint64 num_y, fft::Type type, in Create2dPlanWithScratchAllocator() 356 uint64 num_y, uint64 num_z, in Create3dPlan() 373 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 | 404 uint64 num_y, fft::Type type, in Create2dPlan() 418 Stream *stream, uint64 num_x, uint64 num_y, fft::Type type, in Create2dPlanWithScratchAllocator() 433 uint64 num_y, uint64 num_z, in Create3dPlan() 448 Stream *stream, uint64 num_x, uint64 num_y, uint64 num_z, fft::Type type, in Create3dPlanWithScratchAllocator()
|
/external/tensorflow/tensorflow/cc/framework/ |
D | gradients.cc | 449 const int num_y = n->num_outputs(); in AddGradients() local
|
/external/tensorflow/tensorflow/core/common_runtime/ |
D | function.cc | 2021 const int num_y = static_cast<int>(gbody_->ret_nodes.size()); in Compute() local
|