Searched defs:num_z (Results 1 – 5 of 5) sorted by relevance
| /external/mesa3d/src/panfrost/lib/ |
| D | pan_invocation.c | 47 unsigned num_z, in panfrost_pack_work_groups_compute()
|
| /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 | 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 | 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()
|