Searched refs:transformed_input (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | cudnn_pooling_gpu.cc | 47 Tensor transformed_input; in Compute() local
53 &transformed_input)); in Compute()
56 transformed_input.tensor<T, 5>()); in Compute()
58 transformed_input = tensor_in; in Compute()
92 auto input_data = AsDeviceMemory(transformed_input.template flat<T>().data(), in Compute()
93 transformed_input.template flat<T>().size()); in Compute()
133 Tensor transformed_input; in Compute() local
140 &transformed_input)); in Compute()
142 transformed_input = *tensor_in; in Compute()
177 transformed_input.tensor<T, 5>()); in Compute()
[all …]
|
| D | pooling_ops_common.cc | 179 Tensor transformed_input; in Compute() local
185 &transformed_input)); in Compute()
188 transformed_input.tensor<T, 4>()); in Compute()
190 transformed_input = tensor_in; in Compute()
204 auto& transformed_input = tensor_in; in Compute() local
254 transformed_input.template flat<T>().data()), in Compute()
255 transformed_input.template flat<T>().size()); in Compute()
311 Tensor transformed_input; in Compute() local
318 &transformed_input)); in Compute()
320 transformed_input = *tensor_in; in Compute()
[all …]
|
| D | conv_ops_3d.cc | 309 Tensor transformed_input; in launch() local
315 &transformed_input)); in launch()
320 To32Bit(transformed_input.tensor<T, 5>()), data_format); in launch()
321 input = transformed_input; in launch()
332 Tensor transformed_input; in launch() local
334 nchw_shape, &transformed_input)); in launch()
341 transformed_input.tensor<T, 5>()); in launch()
342 input = transformed_input; in launch()
|
| D | conv_ops_fused_impl.h | 624 Tensor transformed_input;
632 &transformed_input));
637 {{rows_odd, cols_odd}}, To32Bit(transformed_input.tensor<T, 4>()),
640 input = transformed_input;
651 Tensor transformed_input;
654 nchw_shape, &transformed_input));
658 transformed_input.tensor<T, 4>());
659 input = transformed_input;
|
| D | conv_ops.cc | 705 Tensor transformed_input; in operator ()() local
714 &transformed_input)); in operator ()()
734 To32Bit(transformed_input.tensor<T, 4>()), data_format); in operator ()()
736 input = transformed_input; in operator ()()
746 Tensor transformed_input; in operator ()() local
748 nchw_shape, &transformed_input)); in operator ()()
752 transformed_input.tensor<T, 4>()); in operator ()()
753 input = transformed_input; in operator ()()
|
| D | conv_grad_filter_ops.cc | 784 Tensor transformed_input; in operator ()() local
793 nchw_shape, &transformed_input)); in operator ()()
797 transformed_input.tensor<T, 4>()); in operator ()()
800 CHECK(transformed_input.CopyFrom(compatible_input, nchw_shape)); in operator ()()
803 transformed_input = compatible_input; in operator ()()
812 auto input_ptr = AsDeviceMemory(transformed_input.template flat<T>().data(), in operator ()()
813 transformed_input.template flat<T>().size()); in operator ()()
876 LogConvAutotuneResults(ctx->op_kernel().def(), transformed_input, in operator ()()
|
| D | conv_grad_ops_3d.cc | 1711 Tensor transformed_input; in Compute() local
1719 nchw_shape, &transformed_input)); in Compute()
1723 transformed_input.tensor<T, 5>()); in Compute()
1725 CHECK(transformed_input.CopyFrom(compatible_input, nchw_shape)); in Compute()
1728 transformed_input = compatible_input; in Compute()
1738 AsDeviceMemory(transformed_input.template flat<T>().data(), in Compute()
1739 transformed_input.template flat<T>().size()); in Compute()
|
| /external/tensorflow/tensorflow/contrib/fused_conv/kernels/ |
| D | fused_conv2d_bias_activation_op.cc | 509 Tensor transformed_input; in launch() local
|