| /external/tensorflow/tensorflow/core/kernels/ |
| cudnn_pooling_gpu.cc | 46 Tensor transformed_input; local
52 &transformed_input));
55 transformed_input.tensor<T, 5>());
57 transformed_input = tensor_in;
91 auto input_data = AsDeviceMemory(transformed_input.template flat<T>().data(),
92 transformed_input.template flat<T>().size());
133 Tensor transformed_input; local
140 &transformed_input));
142 transformed_input = *tensor_in;
177 transformed_input.tensor<T, 5>())
[all...] |
| pooling_ops_common.cc | 162 Tensor transformed_input; local
168 &transformed_input));
171 transformed_input.tensor<T, 4>());
173 transformed_input = tensor_in;
211 auto input_data = AsDeviceMemory(transformed_input.template flat<T>().data(),
212 transformed_input.template flat<T>().size());
265 Tensor transformed_input; local
272 &transformed_input));
274 transformed_input = *tensor_in;
314 transformed_input.tensor<T, 4>())
[all...] |
| conv_ops_3d.cc | 270 Tensor transformed_input; local
276 &transformed_input));
281 To32Bit(transformed_input.tensor<T, 5>()), data_format);
282 input = transformed_input;
293 Tensor transformed_input; local
295 nchw_shape, &transformed_input));
302 transformed_input.tensor<T, 5>());
303 input = transformed_input;
|
| conv_ops.cc | 588 Tensor transformed_input; local
596 &transformed_input));
601 To32Bit(transformed_input.tensor<T, 4>()), data_format);
603 input = transformed_input;
614 Tensor transformed_input; local
616 nchw_shape, &transformed_input));
620 transformed_input.tensor<T, 4>());
621 input = transformed_input;
[all...] |
| conv_grad_filter_ops.cc | 854 Tensor transformed_input; local
[all...] |
| conv_grad_ops_3d.cc | 976 Tensor transformed_input; variable
[all...] |
| /external/tensorflow/tensorflow/contrib/fused_conv/kernels/ |
| fused_conv2d_bias_activation_op.cc | 374 Tensor transformed_input; local
|
