Searched refs:bias_op (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.cc | 972 miopenFusionOpDescriptor_t bias_op; in SetBiasArgs() local
973 auto status = wrap::miopenFusionPlanGetOp(fusion_plan_, op_idx, &bias_op); in SetBiasArgs()
979 status = wrap::miopenSetOpArgsBiasForward(fusion_args_, bias_op, alpha, in SetBiasArgs()
1140 miopenFusionOpDescriptor_t bias_op; in ScopedFusionPlanConvolutionBiasActivation() local
1141 status = wrap::miopenCreateOpBiasForward(fusion_plan_, &bias_op, in ScopedFusionPlanConvolutionBiasActivation()
|
| /external/tensorflow/tensorflow/contrib/makefile/ |
| D | tf_op_files.txt | 15 tensorflow/core/kernels/bias_op.cc
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | BUILD | 3859 ":bias_op",
3912 name = "bias_op",
3913 prefix = "bias_op",
5457 "bias_op.cc",
5458 "bias_op.h",
6844 ":bias_op",
|