Searched refs:transformed_out_backprop (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | conv_grad_input_ops.cc | 313 Tensor transformed_out_backprop; in operator ()() local
322 &transformed_out_backprop)); in operator ()()
325 transformed_out_backprop.tensor<T, 4>()); in operator ()()
328 CHECK(transformed_out_backprop.CopyFrom(out_backprop, compute_shape)); in operator ()()
331 transformed_out_backprop = out_backprop; in operator ()()
347 AsDeviceMemory(transformed_out_backprop.template flat<T>().data(), in operator ()()
348 transformed_out_backprop.template flat<T>().size()); in operator ()()
|
| D | conv_grad_ops_3d.cc | 1353 Tensor transformed_out_backprop; in Compute() local
1361 &transformed_out_backprop)); in Compute()
1364 transformed_out_backprop.tensor<T, 5>()); in Compute()
1366 CHECK(transformed_out_backprop.CopyFrom(out_backprop, nchw_shape)); in Compute()
1369 transformed_out_backprop = out_backprop; in Compute()
1385 AsDeviceMemory(transformed_out_backprop.template flat<T>().data(), in Compute()
1386 transformed_out_backprop.template flat<T>().size()); in Compute()
1843 Tensor transformed_out_backprop; in Compute() local
1851 &transformed_out_backprop)); in Compute()
1855 transformed_out_backprop.tensor<T, 5>()); in Compute()
[all …]
|
| D | conv_grad_filter_ops.cc | 898 Tensor transformed_out_backprop; in operator ()() local
907 &transformed_out_backprop)); in operator ()()
910 transformed_out_backprop.tensor<T, 4>()); in operator ()()
913 CHECK(transformed_out_backprop.CopyFrom(out_backprop, compute_shape)); in operator ()()
916 transformed_out_backprop = out_backprop; in operator ()()
944 AsDeviceMemory(transformed_out_backprop.template flat<T>().data(), in operator ()()
945 transformed_out_backprop.template flat<T>().size()); in operator ()()
|