| /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()
89 int num_x, int num_y) { in ReduceToScalar()
100 const string& device, const string& reduce, int num_x, in DoRowReduce()
112 const string& device, const string& reduce, int num_x, in DoColReduce()
124 const string& device, const string& reduce, int num_x, in Do3DYReduce()
136 const string& device, const string& reduce, int num_x, in Do3DXZReduce()
148 const int num_x = state.range(0); in BM_Sum2DToScalarGPU() local
156 const int num_x = state.range(0); in BM_Sum2DToScalarGPUComplex() local
[all …]
|
| D | scan_ops_test.cc | 25 static Graph* LargeOneDCumsum(int num_x, bool reverse = false) { in LargeOneDCumsum()
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()
71 const string& device, int num_x, in LargeOneDimensional()
82 const string& device, int num_x, int num_y, in DoRowCumsum()
94 const string& device, int num_x, int num_y, in DoColCumsum()
106 const string& device, int num_x, int num_y, in Do3DYCumsum()
118 const int num_x = state.range(0); in BM_OneDCumsumGPU() local
125 const int num_x = state.range(0); in BM_OneDCumsumGPUHalf() local
132 const int num_x = state.range(0); in BM_Sum2DRowCumsumGPU() local
[all …]
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_fft.cc | 371 std::unique_ptr<fft::Plan> ROCMFft::Create1dPlan(Stream *stream, uint64 num_x, in Create1dPlan()
388 Stream *stream, uint64 num_x, fft::Type type, bool in_place_fft, in Create1dPlanWithScratchAllocator()
402 std::unique_ptr<fft::Plan> ROCMFft::Create2dPlan(Stream *stream, uint64 num_x, in Create2dPlan()
417 Stream *stream, uint64 num_x, uint64 num_y, fft::Type type, in Create2dPlanWithScratchAllocator()
431 std::unique_ptr<fft::Plan> ROCMFft::Create3dPlan(Stream *stream, uint64 num_x, in Create3dPlan()
447 Stream *stream, uint64 num_x, uint64 num_y, uint64 num_z, fft::Type type, in Create3dPlanWithScratchAllocator()
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_fft.cc | 294 std::unique_ptr<fft::Plan> CUDAFft::Create1dPlan(Stream *stream, uint64 num_x, in Create1dPlan()
312 Stream *stream, uint64 num_x, fft::Type type, bool in_place_fft, in Create1dPlanWithScratchAllocator()
327 std::unique_ptr<fft::Plan> CUDAFft::Create2dPlan(Stream *stream, uint64 num_x, in Create2dPlan()
343 Stream *stream, uint64 num_x, uint64 num_y, fft::Type type, in Create2dPlanWithScratchAllocator()
358 std::unique_ptr<fft::Plan> CUDAFft::Create3dPlan(Stream *stream, uint64 num_x, in Create3dPlan()
376 Stream *stream, uint64 num_x, uint64 num_y, uint64 num_z, fft::Type type, in Create3dPlanWithScratchAllocator()
|
| /external/mesa3d/src/panfrost/lib/ |
| D | pan_invocation.c | 45 unsigned num_x, in panfrost_pack_work_groups_compute()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | gradients.cc | 103 const int num_x = n->num_inputs(); in AddSymGrad() local
354 const int num_x = n->num_inputs(); in Compute() local
|
| D | function.cc | 1469 const size_t num_x = fbody_->arg_nodes.size(); in Compute() local
|