| /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() argument
29 Tensor data(DataTypeToEnum<T>::value, TensorShape({num_x, num_y})); in ToScalar()
39 static Graph* ColReduce(const string& reduce, int num_x, int num_y) { in ColReduce() argument
41 Tensor data(DT_FLOAT, TensorShape({num_x, num_y})); in ColReduce()
50 static Graph* RowReduce(const string& reduce, int num_x, int num_y) { in RowReduce() argument
52 Tensor data(DT_FLOAT, TensorShape({num_x, num_y})); in RowReduce()
89 int num_x, int num_y) { in ReduceToScalar() argument
90 test::Benchmark(device, ToScalar<T>(reduce, num_x, num_y), in ReduceToScalar()
93 state.SetItemsProcessed(static_cast<int64>(state.iterations()) * num_x * in ReduceToScalar()
95 state.SetBytesProcessed(static_cast<int64>(state.iterations()) * num_x * in ReduceToScalar()
[all …]
|
| D | scan_ops_test.cc | 25 static Graph* LargeOneDCumsum(int num_x, bool reverse = false) { in LargeOneDCumsum() argument
27 Tensor data(DataTypeToEnum<T>::value, TensorShape({num_x})); in LargeOneDCumsum()
36 static Graph* ColCumsum(int num_x, int num_y, bool reverse = false) { in ColCumsum() argument
38 Tensor data(DT_FLOAT, TensorShape({num_x, num_y})); in ColCumsum()
47 static Graph* RowCumsum(int num_x, int num_y, bool reverse = false) { in RowCumsum() argument
49 Tensor data(DT_FLOAT, TensorShape({num_x, num_y})); in RowCumsum()
71 const string& device, int num_x, in LargeOneDimensional() argument
73 test::Benchmark(device, LargeOneDCumsum<T>(num_x, reverse), in LargeOneDimensional()
76 state.SetItemsProcessed(static_cast<int64>(state.iterations()) * num_x); in LargeOneDimensional()
77 state.SetBytesProcessed(static_cast<int64>(state.iterations()) * num_x * in LargeOneDimensional()
[all …]
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | fft.h | 94 virtual std::unique_ptr<Plan> Create1dPlan(Stream *stream, uint64 num_x,
98 virtual std::unique_ptr<Plan> Create2dPlan(Stream *stream, uint64 num_x,
103 virtual std::unique_ptr<Plan> Create3dPlan(Stream *stream, uint64 num_x,
109 Stream *stream, uint64 num_x, Type type, bool in_place_fft,
114 Stream *stream, uint64 num_x, uint64 num_y, Type type, bool in_place_fft,
119 Stream *stream, uint64 num_x, uint64 num_y, uint64 num_z, Type type,
214 std::unique_ptr<fft::Plan> Create1dPlan(Stream *stream, uint64 num_x, \
217 std::unique_ptr<fft::Plan> Create2dPlan(Stream *stream, uint64 num_x, \
221 Stream *stream, uint64 num_x, uint64 num_y, uint64 num_z, \
224 Stream *stream, uint64 num_x, fft::Type type, bool in_place_fft, \
[all …]
|
| /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() argument
298 uint64 elem_count[1] = {num_x}; in Create1dPlan()
304 LOG(ERROR) << "Plan Parameters: num_x: " << num_x; in Create1dPlan()
312 Stream *stream, uint64 num_x, fft::Type type, bool in_place_fft, in Create1dPlanWithScratchAllocator() argument
315 uint64 elem_count[1] = {num_x}; in Create1dPlanWithScratchAllocator()
319 LOG(ERROR) << "Plan Parameters: num_x: " << num_x; in Create1dPlanWithScratchAllocator()
327 std::unique_ptr<fft::Plan> CUDAFft::Create2dPlan(Stream *stream, uint64 num_x, in Create2dPlan() argument
331 uint64 elem_count[2] = {num_x, num_y}; in Create2dPlan()
335 LOG(ERROR) << "Plan Parameters: num_x: " << num_x << " num_y: " << num_y; in Create2dPlan()
343 Stream *stream, uint64 num_x, uint64 num_y, fft::Type type, in Create2dPlanWithScratchAllocator() argument
[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() argument
375 uint64 elem_count[1] = {num_x}; in Create1dPlan()
388 Stream *stream, uint64 num_x, fft::Type type, bool in_place_fft, in Create1dPlanWithScratchAllocator() argument
391 uint64 elem_count[1] = {num_x}; in Create1dPlanWithScratchAllocator()
402 std::unique_ptr<fft::Plan> ROCMFft::Create2dPlan(Stream *stream, uint64 num_x, in Create2dPlan() argument
406 uint64 elem_count[2] = {num_x, num_y}; in Create2dPlan()
417 Stream *stream, uint64 num_x, uint64 num_y, fft::Type type, in Create2dPlanWithScratchAllocator() argument
420 uint64 elem_count[2] = {num_x, num_y}; in Create2dPlanWithScratchAllocator()
431 std::unique_ptr<fft::Plan> ROCMFft::Create3dPlan(Stream *stream, uint64 num_x, in Create3dPlan() argument
436 uint64 elem_count[3] = {num_x, num_y, num_z}; in Create3dPlan()
[all …]
|
| /external/mesa3d/src/panfrost/lib/ |
| D | pan_invocation.c | 45 unsigned num_x, in panfrost_pack_work_groups_compute() argument
64 MALI_POSITIVE(num_x), in panfrost_pack_work_groups_compute()
|
| D | pan_encoder.h | 38 unsigned num_x,
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | gradients.cc | 103 const int num_x = n->num_inputs(); in AddSymGrad() local
112 std::vector<NodeOut> n_inputs(num_x); in AddSymGrad()
126 CHECK_EQ(ndef.input_size(), num_x + num_y); in AddSymGrad()
354 const int num_x = n->num_inputs(); in Compute() local
390 graph_->AddEdge(dy[i].node, dy[i].index, grad, num_x + i); in Compute()
|
| D | function.cc | 1469 const size_t num_x = fbody_->arg_nodes.size(); in Compute() local
1471 x_node_outputs.reserve(num_x); in Compute()
|