Home
last modified time | relevance | path

Searched defs:num_z (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dreduction_ops_test.cc61 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()
H A Dscan_ops_test.cc58 static Graph* ThreeDYCumsum(int num_y, int num_z, bool reverse = false) { in ThreeDYCumsum()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/
H A Dcuda_fft.cc381 uint64_t num_y, uint64 num_z, in Create3dPlan()
399 Stream *stream, uint64_t num_x, uint64 num_y, uint64 num_z, fft::Type type, in Create3dPlanWithScratchAllocator()
/aosp_15_r20/external/tensorflow/tensorflow/stream_executor/rocm/
H A Drocm_fft.cc434 uint64_t num_y, uint64 num_z, in Create3dPlan()
449 Stream *stream, uint64_t num_x, uint64 num_y, uint64 num_z, fft::Type type, in Create3dPlanWithScratchAllocator()
/aosp_15_r20/external/mesa3d/src/panfrost/lib/
H A Dpan_encoder.h152 unsigned num_z, unsigned size_x, in panfrost_pack_work_groups_compute()