Searched refs:input_param (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | conv_ops_3d.cc | 204 const Tensor& input_param, const Tensor& filter, in launch()
211 Tensor input = input_param; in launch()
318 ctx->eigen_device<GPUDevice>(), To32Bit(input_param.tensor<T, 5>()), in launch()
|
| D | conv_ops_fused_impl.h | 556 bool cudnn_use_autotune, const Tensor& input_param,
579 Tensor input = input_param;
636 To32Bit(input_param.tensor<T, 4>()), {{0, 0}},
|
| D | conv_ops.cc | 580 const Tensor& input_param, const Tensor& filter, int row_dilation, in operator ()() argument
597 Tensor input = input_param; in operator ()()
730 ctx->eigen_device<GPUDevice>(), To32Bit(input_param.tensor<T, 4>()), in operator ()()
|