Searched refs:num_z (Results 1 – 7 of 7) sorted by relevance
| /external/mesa3d/src/panfrost/lib/ |
| D | pan_invocation.c | 47 unsigned num_z, in panfrost_pack_work_groups_compute() argument
66 MALI_POSITIVE(num_z), in panfrost_pack_work_groups_compute()
92 if (quirk_graphics && (num_z <= 1)) in panfrost_pack_work_groups_compute()
|
| D | pan_encoder.h | 40 unsigned num_z,
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | fft.h | 104 uint64 num_y, uint64 num_z,
119 Stream *stream, uint64 num_x, uint64 num_y, uint64 num_z, Type type,
221 Stream *stream, uint64 num_x, uint64 num_y, uint64 num_z, \
230 Stream *stream, uint64 num_x, uint64 num_y, uint64 num_z, \
|
| /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() argument
63 Tensor data(DT_FLOAT, TensorShape({4, num_y, num_z})); in ThreeDYReduce()
72 static Graph* ThreeDXZReduce(const string& reduce, int num_y, int num_z) { in ThreeDXZReduce() argument
74 Tensor data(DT_FLOAT, TensorShape({4, num_y, num_z})); in ThreeDXZReduce()
|
| D | scan_ops_test.cc | 58 static Graph* ThreeDYCumsum(int num_y, int num_z, bool reverse = false) { in ThreeDYCumsum() argument
60 Tensor data(DT_FLOAT, TensorShape({32, num_y, num_z})); in ThreeDYCumsum()
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_fft.cc | 359 uint64 num_y, uint64 num_z, in Create3dPlan() argument
363 uint64 elem_count[3] = {num_x, num_y, num_z}; in Create3dPlan()
368 << " num_z: " << num_z; in Create3dPlan()
376 Stream *stream, uint64 num_x, uint64 num_y, uint64 num_z, fft::Type type, in Create3dPlanWithScratchAllocator() argument
379 uint64 elem_count[3] = {num_x, num_y, num_z}; in Create3dPlanWithScratchAllocator()
384 << " num_z: " << num_z; in Create3dPlanWithScratchAllocator()
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_fft.cc | 432 uint64 num_y, uint64 num_z, in Create3dPlan() argument
436 uint64 elem_count[3] = {num_x, num_y, num_z}; in Create3dPlan()
447 Stream *stream, uint64 num_x, uint64 num_y, uint64 num_z, fft::Type type, in Create3dPlanWithScratchAllocator() argument
450 uint64 elem_count[3] = {num_x, num_y, num_z}; in Create3dPlanWithScratchAllocator()
|