Searched defs:side_input_data (Results 1 – 5 of 5) 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/ |
| D | stream.h | 393 const DeviceMemory<SideInputT> &side_input_data, ScaleT side_input_scale, in FusedConvolveWithAlgorithm()
421 const DeviceMemory<SideInputT> &side_input_data, ScaleT side_input_scale, in FusedConvolveWithExecutionPlan()
|
| D | dnn.h | 1260 DeviceMemoryBase side_input_data, double side_input_scale, in DoFusedConvolve()
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.cc | 4875 DeviceMemoryBase side_input_data, double side_input_scale, in DoFusedConvolve()
5022 DeviceMemoryBase side_input_data, double side_input_scale, in DoFusedConvolveWithExecutionPlan()
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.cc | 3678 DeviceMemoryBase side_input_data, double side_input_scale, in DoFusedConvolve()
|