| /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()
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()
[all …]
|
| D | scan_ops_test.cc | 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()
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()
82 const string& device, int num_x, int num_y, in DoRowCumsum() argument
84 test::Benchmark(device, RowCumsum(num_x, num_y, reverse), in DoRowCumsum()
88 num_y); in DoRowCumsum()
90 num_y * sizeof(float)); in DoRowCumsum()
[all …]
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | fft.h | 99 uint64 num_y, Type type,
104 uint64 num_y, uint64 num_z,
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,
218 uint64 num_y, fft::Type type, \
221 Stream *stream, uint64 num_x, uint64 num_y, uint64 num_z, \
227 Stream *stream, uint64 num_x, uint64 num_y, fft::Type type, \
230 Stream *stream, uint64 num_x, uint64 num_y, uint64 num_z, \
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | gradients.cc | 104 const int num_y = n->num_outputs(); in AddSymGrad() local
105 CHECK_EQ(num_y, grads.size()); in AddSymGrad()
126 CHECK_EQ(ndef.input_size(), num_x + num_y); in AddSymGrad()
291 const int num_y = y_grad_node_outputs_.size(); in InitBackprop() local
292 for (int i = 0; i < num_y; ++i) { in InitBackprop()
355 const int num_y = n->num_outputs(); in Compute() local
361 CHECK_EQ(1, num_y); in Compute()
367 dy.resize(num_y, {nullptr, 0}); in Compute()
368 for (int i = 0; i < num_y; ++i) { in Compute()
389 for (int i = 0; i < num_y; ++i) { in Compute()
|
| D | function.cc | 1446 const int num_y = static_cast<int>(gbody->ret_nodes.size()); in Compute() local
1453 y_node_outputs.reserve(num_y); in Compute()
1455 y_grad_node_outputs.reserve(num_y); in Compute()
1456 for (int i = 0; i < num_y; ++i) { in Compute()
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_fft.cc | 328 uint64 num_y, fft::Type type, 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
346 uint64 elem_count[2] = {num_x, num_y}; in Create2dPlanWithScratchAllocator()
350 LOG(ERROR) << "Plan Parameters: num_x: " << num_x << " num_y: " << num_y; in Create2dPlanWithScratchAllocator()
359 uint64 num_y, uint64 num_z, in Create3dPlan() argument
363 uint64 elem_count[3] = {num_x, num_y, num_z}; in Create3dPlan()
367 LOG(ERROR) << "Plan Parameters: num_x: " << num_x << " num_y: " << num_y in Create3dPlan()
376 Stream *stream, uint64 num_x, uint64 num_y, uint64 num_z, fft::Type type, in Create3dPlanWithScratchAllocator() argument
[all …]
|
| /external/mesa3d/src/panfrost/lib/ |
| D | pan_invocation.c | 46 unsigned num_y, in panfrost_pack_work_groups_compute() argument
65 MALI_POSITIVE(num_y), in panfrost_pack_work_groups_compute()
|
| D | pan_encoder.h | 39 unsigned num_y,
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_fft.cc | 403 uint64 num_y, fft::Type type, 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()
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()
|
| /external/tensorflow/tensorflow/cc/framework/ |
| D | gradients.cc | 450 const int num_y = n->num_outputs(); in AddGradients() local
452 dy.resize(num_y, {nullptr, 0}); in AddGradients()
454 for (int i = 0; i < num_y; ++i) { in AddGradients()
493 if (IsPrimitiveOpWithNoGrad(n->type_string()) || num_no_grad == num_y) { in AddGradients()
504 if (num_no_grad > 0 && num_no_grad < num_y) { in AddGradients()
|