Home
last modified time | relevance | path

Searched refs:side_input_data (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dfused_batch_norm_ex_op_test.cc72 const Tensor& side_input_data, FusedBatchNormOutputs* forward,
139 const Tensor& var_data, const Tensor& side_input_data, in RunFusedBatchNorm() argument
160 Input::Initializer(side_input_data)); in RunFusedBatchNorm()
228 const Tensor& side_input_data, in RunFusedBatchNormEx() argument
252 Input::Initializer(side_input_data)); in RunFusedBatchNormEx()
435 const Tensor& side_input_data, FusedBatchNormOutputs* fwd, in VerifyFusedBatchNormEx()
439 side_input_data, data_format, is_training, in VerifyFusedBatchNormEx()
447 const Tensor& side_input_data, FusedBatchNormOutputs* fwd, in VerifyFusedBatchNormEx() argument
451 side_input_data, data_format, is_training, in VerifyFusedBatchNormEx()
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.h286 const DeviceMemory<double>& side_input_data, double side_input_scale,
300 const DeviceMemory<float>& side_input_data, float side_input_scale,
314 const DeviceMemory<Eigen::half>& side_input_data, float side_input_scale,
330 const DeviceMemory<int8>& side_input_data, float side_input_scale,
344 const DeviceMemory<float>& side_input_data, float side_input_scale,
623 const DeviceMemory<OutputType>& side_input_data,
Dcuda_dnn.cc3183 const DeviceMemory<OutputType>& side_input_data, ScaleType side_input_scale, in DoFusedConvolveImpl() argument
3242 : side_input_data.opaque(); in DoFusedConvolveImpl()
3762 const DeviceMemory<double>& side_input_data, double side_input_scale, in DoFusedConvolve() argument
3771 filter_descriptor, filter_data, convolution_descriptor, side_input_data, in DoFusedConvolve()
3784 const DeviceMemory<float>& side_input_data, float side_input_scale, in DoFusedConvolve() argument
3793 filter_descriptor, filter_data, convolution_descriptor, side_input_data, in DoFusedConvolve()
3806 const DeviceMemory<Eigen::half>& side_input_data, float side_input_scale, in DoFusedConvolve() argument
3816 filter_descriptor, filter_data, convolution_descriptor, side_input_data, in DoFusedConvolve()
3829 const DeviceMemory<int8>& side_input_data, float side_input_scale, in DoFusedConvolve() argument
3847 filter_descriptor, filter_data, convolution_descriptor, side_input_data, in DoFusedConvolve()
[all …]
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.h324 const DeviceMemory<double>& side_input_data, double side_input_scale,
338 const DeviceMemory<float>& side_input_data, float side_input_scale,
352 const DeviceMemory<Eigen::half>& side_input_data, float side_input_scale,
368 const DeviceMemory<int8>& side_input_data, float side_input_scale,
Drocm_dnn.cc3681 const DeviceMemory<double>& side_input_data, double side_input_scale, in DoFusedConvolve() argument
3697 const DeviceMemory<float>& side_input_data, float side_input_scale, in DoFusedConvolve() argument
3713 const DeviceMemory<Eigen::half>& side_input_data, float side_input_scale, in DoFusedConvolve() argument
3730 const DeviceMemory<int8>& side_input_data, float side_input_scale, in DoFusedConvolve() argument
/external/tensorflow/tensorflow/stream_executor/
Ddnn.h1149 const DeviceMemory<double>& side_input_data, double side_input_scale, in DoFusedConvolve() argument
1167 const DeviceMemory<float>& side_input_data, float side_input_scale, in DoFusedConvolve() argument
1186 const DeviceMemory<Eigen::half>& side_input_data, float side_input_scale, in DoFusedConvolve() argument
1207 const DeviceMemory<int8>& side_input_data, float side_input_scale, in DoFusedConvolve() argument
Dstream.cc452 const DeviceMemory<double> &side_input_data, double side_input_scale, in FusedConvolveWithAlgorithm() argument
462 PARAM(side_input_data), PARAM(side_input_scale), in FusedConvolveWithAlgorithm()
469 filter_descriptor, filter_data, convolution_descriptor, side_input_data, in FusedConvolveWithAlgorithm()
483 const DeviceMemory<float> &side_input_data, float side_input_scale, in FusedConvolveWithAlgorithm() argument
493 PARAM(side_input_data), PARAM(side_input_scale), in FusedConvolveWithAlgorithm()
500 filter_descriptor, filter_data, convolution_descriptor, side_input_data, in FusedConvolveWithAlgorithm()
514 const DeviceMemory<Eigen::half> &side_input_data, float side_input_scale, in FusedConvolveWithAlgorithm() argument
525 PARAM(side_input_data), PARAM(side_input_scale), in FusedConvolveWithAlgorithm()
532 filter_descriptor, filter_data, convolution_descriptor, side_input_data, in FusedConvolveWithAlgorithm()
546 const DeviceMemory<int8> &side_input_data, float side_input_scale, in FusedConvolveWithAlgorithm() argument
[all …]
Dstream.h360 const DeviceMemory<double> &side_input_data, double side_input_scale,
374 const DeviceMemory<float> &side_input_data, float side_input_scale,
388 const DeviceMemory<Eigen::half> &side_input_data, float side_input_scale,
403 const DeviceMemory<int8> &side_input_data, float side_input_scale,
417 const DeviceMemory<float> &side_input_data, float side_input_scale,