Home
last modified time | relevance | path

Searched defs:num_x (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dreduction_ops_test.cc27 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()
88 const string& reduce, int num_x, int num_y) { in ReduceToScalar()
96 int num_x, int num_y) { in DoRowReduce()
104 int num_x, int num_y) { in DoColReduce()
112 int num_x, int num_y) { in Do3DYReduce()
120 int num_x, int num_y) { in Do3DXZReduce()
127 static void BM_Sum2DToScalarGPU(int iters, int num_x, int num_y) { in BM_Sum2DToScalarGPU()
132 static void BM_Sum2DToScalarGPUComplex(int iters, int num_x, int num_y) { in BM_Sum2DToScalarGPUComplex()
[all …]
Dscan_ops_test.cc25 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()
70 static void LargeOneDimensional(int iters, const string& device, int num_x, in LargeOneDimensional()
77 static void DoRowCumsum(int iters, const string& device, int num_x, int num_y, in DoRowCumsum()
85 static void DoColCumsum(int iters, const string& device, int num_x, int num_y, in DoColCumsum()
93 static void Do3DYCumsum(int iters, const string& device, int num_x, int num_y, in Do3DYCumsum()
101 static void BM_OneDCumsumGPU(int iters, int num_x) { in BM_OneDCumsumGPU()
106 static void BM_OneDCumsumGPUHalf(int iters, int num_x) { in BM_OneDCumsumGPUHalf()
111 static void BM_Sum2DRowCumsumGPU(int iters, int num_x, int num_y) { in BM_Sum2DRowCumsumGPU()
[all …]
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_fft.cc291 std::unique_ptr<fft::Plan> CUDAFft::Create1dPlan(Stream *stream, uint64 num_x, in Create1dPlan()
309 Stream *stream, uint64 num_x, fft::Type type, bool in_place_fft, in Create1dPlanWithScratchAllocator()
324 std::unique_ptr<fft::Plan> CUDAFft::Create2dPlan(Stream *stream, uint64 num_x, in Create2dPlan()
340 Stream *stream, uint64 num_x, uint64 num_y, fft::Type type, in Create2dPlanWithScratchAllocator()
355 std::unique_ptr<fft::Plan> CUDAFft::Create3dPlan(Stream *stream, uint64 num_x, 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/
Drocm_fft.cc372 std::unique_ptr<fft::Plan> ROCMFft::Create1dPlan(Stream *stream, uint64 num_x, in Create1dPlan()
389 Stream *stream, uint64 num_x, fft::Type type, bool in_place_fft, in Create1dPlanWithScratchAllocator()
403 std::unique_ptr<fft::Plan> ROCMFft::Create2dPlan(Stream *stream, uint64 num_x, in Create2dPlan()
418 Stream *stream, uint64 num_x, uint64 num_y, fft::Type type, in Create2dPlanWithScratchAllocator()
432 std::unique_ptr<fft::Plan> ROCMFft::Create3dPlan(Stream *stream, uint64 num_x, in Create3dPlan()
448 Stream *stream, uint64 num_x, uint64 num_y, uint64 num_z, fft::Type type, in Create3dPlanWithScratchAllocator()
/external/tensorflow/tensorflow/core/graph/
Dgradients.cc102 const int num_x = n->num_inputs(); in AddSymGrad() local
345 const int num_x = n->num_inputs(); in Compute() local
/external/tensorflow/tensorflow/core/common_runtime/
Dfunction.cc2044 const size_t num_x = fbody_->arg_nodes.size(); in Compute() local