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()
148 Tensor transformed_input; in Compute() local
155 &transformed_input)); in Compute()
157 transformed_input = *tensor_in; in Compute()
192 transformed_input.tensor<T, 5>()); in Compute()
[all …]
|
| D | pooling_ops_common.cc | 243 Tensor transformed_input; in Compute() local
249 &transformed_input)); in Compute()
252 transformed_input.tensor<T, 4>()); in Compute()
254 transformed_input = tensor_in; in Compute()
268 Tensor transformed_input = tensor_in; in Compute() local
336 const Tensor& const_transformed_input = transformed_input; in Compute()
345 transformed_input = padded_input; in Compute()
378 transformed_input.template flat<T>().data()), in Compute()
379 transformed_input.template flat<T>().size()); in Compute()
471 Tensor transformed_input; in Compute() local
[all …]
|
| D | conv_ops_3d.cc | 321 Tensor transformed_input; in launch() local
327 &transformed_input)); in launch()
332 To32Bit(transformed_input.tensor<T, 5>()), data_format, T{}); in launch()
333 input = transformed_input; in launch()
360 Tensor transformed_input; in launch() local
362 nchw_shape, &transformed_input)); in launch()
369 transformed_input.tensor<T, 5>()); in launch()
370 input = transformed_input; in launch()
|
| D | conv_ops_fused_impl.h | 498 Tensor transformed_input;
510 &transformed_input));
534 To32Bit(transformed_input.tensor<T, 4>()), params.data_format, T{});
535 input = transformed_input;
545 Tensor transformed_input;
548 nchw_shape, &transformed_input));
552 transformed_input.tensor<T, 4>());
553 input = transformed_input;
|
| D | conv_ops.cc | 801 Tensor transformed_input; in operator ()() local
810 &transformed_input)); in operator ()()
830 To32Bit(transformed_input.tensor<T, 4>()), data_format, T{}); in operator ()()
832 input = transformed_input; in operator ()()
843 Tensor transformed_input; in operator ()() local
845 nchw_shape, &transformed_input)); in operator ()()
849 transformed_input.tensor<T, 4>()); in operator ()()
850 input = transformed_input; in operator ()()
|
| D | lrn_op.cc | 239 Tensor transformed_input; in launch()
244 &transformed_input)); in launch()
247 transformed_input.tensor<T, 4>()); in launch()
270 AsDeviceMemory(transformed_input.template flat<T>().data(), in launch()
271 transformed_input.template flat<T>().size()); in launch()
|
| D | conv_grad_filter_ops.cc | 919 Tensor transformed_input; in operator ()() local
930 &transformed_input)); in operator ()()
934 transformed_input.tensor<T, 4>()); in operator ()()
937 CHECK(transformed_input.CopyFrom(compatible_input, compute_shape)); in operator ()()
940 transformed_input = compatible_input; in operator ()()
949 auto input_ptr = AsDeviceMemory(transformed_input.template flat<T>().data(), in operator ()()
950 transformed_input.template flat<T>().size()); in operator ()()
|
| D | conv_grad_ops_3d.cc | 1862 Tensor transformed_input; in Compute() local
1871 nchw_shape, &transformed_input)); in Compute()
1875 transformed_input.tensor<T, 5>()); in Compute()
1877 CHECK(transformed_input.CopyFrom(compatible_input, nchw_shape)); in Compute()
1880 transformed_input = compatible_input; in Compute()
1890 AsDeviceMemory(transformed_input.template flat<T>().data(), in Compute()
1891 transformed_input.template flat<T>().size()); in Compute()
|