Searched defs:side_input_data (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | fused_batch_norm_ex_op_test.cc | 139 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/cuda/ |
| D | cuda_dnn.cc | 3219 const DeviceMemory<OutputType>& side_input_data, ScaleType side_input_scale, in DoFusedConvolveImpl()
3784 const DeviceMemory<double>& side_input_data, double side_input_scale, in DoFusedConvolve()
3808 const DeviceMemory<float>& side_input_data, float side_input_scale, in DoFusedConvolve()
3832 const DeviceMemory<Eigen::half>& side_input_data, float side_input_scale, in DoFusedConvolve()
3857 const DeviceMemory<int8>& side_input_data, float side_input_scale, in DoFusedConvolve()
3890 const DeviceMemory<float>& side_input_data, float side_input_scale, in DoFusedConvolve()
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.cc | 3440 const DeviceMemory<double>& side_input_data, double side_input_scale, in DoFusedConvolve()
3457 const DeviceMemory<float>& side_input_data, float side_input_scale, in DoFusedConvolve()
3474 const DeviceMemory<Eigen::half>& side_input_data, float side_input_scale, in DoFusedConvolve()
3492 const DeviceMemory<int8>& side_input_data, float side_input_scale, in DoFusedConvolve()
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | stream.cc | 466 const DeviceMemory<double> &side_input_data, double side_input_scale, in ThenFusedConvolveWithAlgorithm()
504 const DeviceMemory<float> &side_input_data, float side_input_scale, in ThenFusedConvolveWithAlgorithm()
542 const DeviceMemory<Eigen::half> &side_input_data, float side_input_scale, in ThenFusedConvolveWithAlgorithm()
581 const DeviceMemory<int8> &side_input_data, float side_input_scale, in ThenFusedConvolveWithAlgorithm()
619 const DeviceMemory<float> &side_input_data, float side_input_scale, in ThenFusedConvolveWithAlgorithm()
|