/external/tensorflow/tensorflow/core/kernels/ |
reduction_ops_test.cc | 27 static Graph* ToScalar(const string& reduce, int num_x, int num_y) { 29 Tensor data(DataTypeToEnum<T>::value, TensorShape({num_x, num_y})); 39 static Graph* ColReduce(const string& reduce, int num_x, int num_y) { 41 Tensor data(DT_FLOAT, TensorShape({num_x, num_y})); 50 static Graph* RowReduce(const string& reduce, int num_x, int num_y) { 52 Tensor data(DT_FLOAT, TensorShape({num_x, num_y})); 61 static Graph* ThreeDYReduce(const string& reduce, int num_y, int num_z) { 63 Tensor data(DT_FLOAT, TensorShape({4, num_y, num_z})); 72 static Graph* ThreeDXZReduce(const string& reduce, int num_y, int num_z) { 74 Tensor data(DT_FLOAT, TensorShape({4, num_y, num_z})) 89 testing::ItemsProcessed(static_cast<int64>(iters) * num_x * num_y); local 97 testing::ItemsProcessed(static_cast<int64>(iters) * num_x * num_y); local 105 testing::ItemsProcessed(static_cast<int64>(iters) * num_x * num_y); local 113 testing::ItemsProcessed(static_cast<int64>(iters) * num_x * num_y); local 121 testing::ItemsProcessed(static_cast<int64>(iters) * num_x * num_y); local [all...] |
scan_ops_test.cc | 36 static Graph* ColCumsum(int num_x, int num_y, bool reverse = false) { 38 Tensor data(DT_FLOAT, TensorShape({num_x, num_y})); 47 static Graph* RowCumsum(int num_x, int num_y, bool reverse = false) { 49 Tensor data(DT_FLOAT, TensorShape({num_x, num_y})); 58 static Graph* ThreeDYCumsum(int num_y, int num_z, bool reverse = false) { 60 Tensor data(DT_FLOAT, TensorShape({32, num_y, num_z})); 77 static void DoRowCumsum(int iters, const string& device, int num_x, int num_y, 79 testing::ItemsProcessed(static_cast<int64>(iters) * num_x * num_y); local 80 testing::BytesProcessed(static_cast<int64>(iters) * num_x * num_y * 82 test::Benchmark(device, RowCumsum(num_x, num_y, reverse)).Run(iters) 87 testing::ItemsProcessed(static_cast<int64>(iters) * num_x * num_y); local 95 testing::ItemsProcessed(static_cast<int64>(iters) * num_x * num_y); local [all...] |
/external/tensorflow/tensorflow/stream_executor/ |
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/graph/ |
gradients.cc | 103 const int num_y = n->num_outputs(); local 104 CHECK_EQ(num_y, grads.size()); 110 // The gradient node should have num_x + num_y inputs. 125 CHECK_EQ(ndef.input_size(), num_x + num_y); 282 const int num_y = y_grad_node_outputs_.size(); local 283 for (int i = 0; i < num_y; ++i) { 344 // "n" has num_x inputs and num_y outputs. 346 const int num_y = n->num_outputs(); local 352 CHECK_EQ(1, num_y); 358 dy.resize(num_y, {nullptr, 0}) [all...] |
/external/tensorflow/tensorflow/stream_executor/cuda/ |
cuda_fft.cc | 325 uint64 num_y, fft::Type type, 328 uint64 elem_count[2] = {num_x, num_y}; 332 LOG(ERROR) << "Plan Parameters: num_x: " << num_x << " num_y: " << num_y; local 340 Stream *stream, uint64 num_x, uint64 num_y, fft::Type type, 343 uint64 elem_count[2] = {num_x, num_y}; 347 LOG(ERROR) << "Plan Parameters: num_x: " << num_x << " num_y: " << num_y; local 356 uint64 num_y, uint64 num_z, 360 uint64 elem_count[3] = {num_x, num_y, num_z} [all...] |
/external/tensorflow/tensorflow/stream_executor/rocm/ |
rocm_fft.cc | 404 uint64 num_y, fft::Type type, 407 uint64 elem_count[2] = {num_x, num_y}; 418 Stream *stream, uint64 num_x, uint64 num_y, fft::Type type, 421 uint64 elem_count[2] = {num_x, num_y}; 433 uint64 num_y, uint64 num_z, 437 uint64 elem_count[3] = {num_x, num_y, num_z}; 448 Stream *stream, uint64 num_x, uint64 num_y, uint64 num_z, fft::Type type, 451 uint64 elem_count[3] = {num_x, num_y, num_z};
|
/external/tensorflow/tensorflow/cc/framework/ |
gradients.cc | 449 const int num_y = n->num_outputs(); local 451 dy.resize(num_y, {nullptr, 0}); 453 for (int i = 0; i < num_y; ++i) { 492 if (IsPrimitiveOpWithNoGrad(n->type_string()) || num_no_grad == num_y) { 503 if (num_no_grad > 0 && num_no_grad < num_y) {
|
/external/tensorflow/tensorflow/core/common_runtime/ |
function.cc | 2021 const int num_y = static_cast<int>(gbody_->ret_nodes.size()); local [all...] |