Searched refs:bias_attr (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_qmatmul_op.cc | 481 dnnl::primitive_attr bias_attr; in GetBiasHandle() local
482 bias_attr.set_output_scales(0, scales); in GetBiasHandle()
492 *input_bias_, *scaled_bias_, bias_attr); in GetBiasHandle()
|
| D | mkl_conv_ops.cc | 1811 dnnl::primitive_attr bias_attr; in GetBiasHandle() local
1813 bias_attr.set_output_scales(0, scales_); in GetBiasHandle()
1815 bias_attr.set_output_scales(1, scales_); in GetBiasHandle()
1838 this->cpu_engine_, scaled_bias_->get_desc(), bias_attr); in GetBiasHandle()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/experimental/tac/transforms/ |
| D | device_transform_patterns.cc | 110 mlir::DenseElementsAttr bias_attr; in EnsureBias() local
113 bias_attr = mlir::DenseFPElementsAttr::get(bias_type, val); in EnsureBias()
120 auto zero_bias = rewriter.create<TFL::ConstOp>(op->getLoc(), bias_attr); in EnsureBias()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| D | prepare_tf.cc | 232 auto bias_attr = rewriter.getZeroAttr(bias_type); in matchAndRewrite() local
234 rewriter.create<TF::ConstOp>(op->getLoc(), bias_type, bias_attr); in matchAndRewrite()
|
| /external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/ |
| D | legalize_tf.cc | 830 auto bias_attr = rewriter.getZeroAttr(bias_type); in matchAndRewrite() local
832 bias_attr.cast<ElementsAttr>()); in matchAndRewrite()
919 auto bias_attr = rewriter.getZeroAttr(bias_type); in matchAndRewrite() local
921 bias_attr.cast<ElementsAttr>()); in matchAndRewrite()
|
| D | legalize_tfl.cc | 1521 DenseElementsAttr bias_attr; in matchAndRewrite() local
1530 bias_attr = in matchAndRewrite()
1549 bias_attr = in matchAndRewrite()
1553 new_bias_type, bias_attr); in matchAndRewrite()
|