Searched refs:transformed_out_backprop (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | conv_grad_filter_ops.cc | 764 Tensor transformed_out_backprop; in operator ()() local
772 &transformed_out_backprop)); in operator ()()
775 transformed_out_backprop.tensor<T, 4>()); in operator ()()
778 CHECK(transformed_out_backprop.CopyFrom(out_backprop, nchw_shape)); in operator ()()
781 transformed_out_backprop = out_backprop; in operator ()()
807 AsDeviceMemory(transformed_out_backprop.template flat<T>().data(), in operator ()()
808 transformed_out_backprop.template flat<T>().size()); in operator ()()
878 transformed_out_backprop, stream->parent(), results); in operator ()()
|
| D | conv_grad_ops_3d.cc | 1300 Tensor transformed_out_backprop; in Compute() local
1308 &transformed_out_backprop)); in Compute()
1311 transformed_out_backprop.tensor<T, 5>()); in Compute()
1313 CHECK(transformed_out_backprop.CopyFrom(out_backprop, nchw_shape)); in Compute()
1316 transformed_out_backprop = out_backprop; in Compute()
1326 AsDeviceMemory(transformed_out_backprop.template flat<T>().data(), in Compute()
1327 transformed_out_backprop.template flat<T>().size()); in Compute()
1693 Tensor transformed_out_backprop; in Compute() local
1700 &transformed_out_backprop)); in Compute()
1704 transformed_out_backprop.tensor<T, 5>()); in Compute()
[all …]
|
| D | conv_grad_input_ops.cc | 884 Tensor transformed_out_backprop; in operator ()() local
892 &transformed_out_backprop)); in operator ()()
895 transformed_out_backprop.tensor<T, 4>()); in operator ()()
898 CHECK(transformed_out_backprop.CopyFrom(out_backprop, nchw_shape)); in operator ()()
901 transformed_out_backprop = out_backprop; in operator ()()
917 AsDeviceMemory(transformed_out_backprop.template flat<T>().data(), in operator ()()
918 transformed_out_backprop.template flat<T>().size()); in operator ()()
988 transformed_filter, transformed_out_backprop, in operator ()()
|