Searched defs:num_z (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | reduction_ops_test.cc | 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()
|
| D | scan_ops_test.cc | 58 static Graph* ThreeDYCumsum(int num_y, int num_z, bool reverse = false) { in ThreeDYCumsum()
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_fft.cc | 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 | 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()
|