Searched refs:side_input_data (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | fused_batch_norm_ex_op_test.cc | 72 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/ |
| D | cuda_dnn.h | 328 DeviceMemoryBase side_input_data, double side_input_scale,
343 DeviceMemoryBase side_input_data, double side_input_scale,
|
| D | cuda_dnn.cc | 4875 DeviceMemoryBase side_input_data, double side_input_scale, in DoFusedConvolve() argument
4948 (side_input_scale == 0) ? output_data.opaque() : side_input_data.opaque(); in DoFusedConvolve()
5022 DeviceMemoryBase side_input_data, double side_input_scale, in DoFusedConvolveWithExecutionPlan() argument
5086 side_input_data.opaque(), biases.opaque(), in DoFusedConvolveWithExecutionPlan()
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | stream.h | 393 const DeviceMemory<SideInputT> &side_input_data, ScaleT side_input_scale, in FusedConvolveWithAlgorithm() argument
406 convolution_descriptor, side_input_data, side_input_scale, in FusedConvolveWithAlgorithm()
421 const DeviceMemory<SideInputT> &side_input_data, ScaleT side_input_scale, in FusedConvolveWithExecutionPlan() argument
435 convolution_descriptor, side_input_data, side_input_scale, in FusedConvolveWithExecutionPlan()
|
| D | dnn.h | 1260 DeviceMemoryBase side_input_data, double side_input_scale, in DoFusedConvolve() argument
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.h | 333 DeviceMemoryBase side_input_data, double side_input_scale,
|
| D | rocm_dnn.cc | 3678 DeviceMemoryBase side_input_data, double side_input_scale, in DoFusedConvolve() argument
|