Searched refs:input_param (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | conv_ops_3d.cc | 213 const Tensor& input_param, const Tensor& filter, in launch()
220 Tensor input = input_param; in launch()
330 ctx->eigen_device<GPUDevice>(), To32Bit(input_param.tensor<T, 5>()), in launch()
|
| D | conv_ops_fused_impl.h | 435 bool cudnn_use_autotune, const Tensor& input_param,
458 Tensor input = input_param;
530 To32Bit(input_param.tensor<T, 4>()),
|
| D | conv_ops.cc | 648 const Tensor& input_param, const Tensor& filter, int row_dilation, in operator ()() argument
665 Tensor input = input_param; in operator ()()
826 ctx->eigen_device<GPUDevice>(), To32Bit(input_param.tensor<T, 4>()), in operator ()()
|
| /external/tensorflow/tensorflow/lite/delegates/nnapi/ |
| D | nnapi_delegate.cc | 2417 auto input_param = context->tensors[node->inputs->data[0]].params; in Validate() local
2419 Expect(input_param.scale == output_param.scale && in Validate()
2420 input_param.zero_point == output_param.zero_point, in Validate()
|