Searched refs:bias_op (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | BUILD | 425 "//tensorflow/core/kernels:bias_op",
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.cc | 1089 miopenFusionOpDescriptor_t bias_op; in SetBiasArgs() local
1090 auto status = wrap::miopenFusionPlanGetOp(fusion_plan_, op_idx, &bias_op); in SetBiasArgs()
1096 status = wrap::miopenSetOpArgsBiasForward(fusion_args_, bias_op, alpha, in SetBiasArgs()
1253 miopenFusionOpDescriptor_t bias_op; in ScopedFusionPlanConvolutionBiasActivation() local
1254 status = wrap::miopenCreateOpBiasForward(fusion_plan_, &bias_op, in ScopedFusionPlanConvolutionBiasActivation()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | BUILD | 1704 ":bias_op",
3881 ":bias_op",
3934 name = "bias_op",
3935 prefix = "bias_op",
5763 "bias_op.cc",
5764 "bias_op.h",
7213 ":bias_op",
|
| /external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/ |
| D | legalize_tfl.cc | 1514 auto bias_op = in matchAndRewrite() local
1516 bias_val = bias_op.getResult(); in matchAndRewrite()
|
| /external/tensorflow/tensorflow/ |
| D | tensorflow.bzl | 2813 dependencies. For eg, see the `//tensorflow/core/kernels:bias_op` target
2816 (for eg. `reduction_ops` dependency for the `bias_op` target above),
|