Home
last modified time | relevance | path

Searched defs:side_input_data (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dfused_batch_norm_ex_op_test.cc139 const Tensor& var_data, const Tensor& side_input_data, in RunFusedBatchNorm()
228 const Tensor& side_input_data, in RunFusedBatchNormEx()
447 const Tensor& side_input_data, FusedBatchNormOutputs* fwd, in VerifyFusedBatchNormEx()
448 FusedBatchNormGradOutputs* bwd) { in VerifyFusedBatchNormEx()
/external/tensorflow/tensorflow/stream_executor/
Ddnn.h1149 const DeviceMemory<double>& side_input_data, double side_input_scale, in DoFusedConvolve()
1167 const DeviceMemory<float>& side_input_data, float side_input_scale, in DoFusedConvolve()
1186 const DeviceMemory<Eigen::half>& side_input_data, float side_input_scale, in DoFusedConvolve()
1207 const DeviceMemory<int8>& side_input_data, float side_input_scale, in DoFusedConvolve()
Dstream.cc452 const DeviceMemory<double> &side_input_data, double side_input_scale, in FusedConvolveWithAlgorithm()
483 const DeviceMemory<float> &side_input_data, float side_input_scale, in FusedConvolveWithAlgorithm()
514 const DeviceMemory<Eigen::half> &side_input_data, float side_input_scale, in FusedConvolveWithAlgorithm()
546 const DeviceMemory<int8> &side_input_data, float side_input_scale, in FusedConvolveWithAlgorithm()
577 const DeviceMemory<float> &side_input_data, float side_input_scale, in FusedConvolveWithAlgorithm()
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.cc3183 const DeviceMemory<OutputType>& side_input_data, ScaleType side_input_scale, in DoFusedConvolveImpl()
3762 const DeviceMemory<double>& side_input_data, double side_input_scale, in DoFusedConvolve()
3784 const DeviceMemory<float>& side_input_data, float side_input_scale, in DoFusedConvolve()
3806 const DeviceMemory<Eigen::half>& side_input_data, float side_input_scale, in DoFusedConvolve()
3829 const DeviceMemory<int8>& side_input_data, float side_input_scale, in DoFusedConvolve()
3860 const DeviceMemory<float>& side_input_data, float side_input_scale, in DoFusedConvolve()
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.cc3681 const DeviceMemory<double>& side_input_data, double side_input_scale, in DoFusedConvolve()
3697 const DeviceMemory<float>& side_input_data, float side_input_scale, in DoFusedConvolve()
3713 const DeviceMemory<Eigen::half>& side_input_data, float side_input_scale, in DoFusedConvolve()
3730 const DeviceMemory<int8>& side_input_data, float side_input_scale, in DoFusedConvolve()