Searched refs:bias_attr (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_qmatmul_op.cc | 473 mkldnn::primitive_attr bias_attr; in GetBiasHandle() local
474 bias_attr.set_output_scales(0, scales); in GetBiasHandle()
484 *input_bias_, *scaled_bias_, bias_attr); in GetBiasHandle()
|
| D | mkl_conv_ops.cc | 1543 mkldnn::primitive_attr bias_attr; in GetBiasHandle() local
1545 bias_attr.set_output_scales(0, scales_); in GetBiasHandle()
1547 bias_attr.set_output_scales(1, scales_); in GetBiasHandle()
1570 this->cpu_engine_, scaled_bias_->get_desc(), bias_attr); in GetBiasHandle()
|
| /external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/ |
| D | legalize_tf.cc | 827 auto bias_attr = rewriter.getZeroAttr(bias_type); in matchAndRewrite() local
829 bias_attr.cast<ElementsAttr>()); in matchAndRewrite()
916 auto bias_attr = rewriter.getZeroAttr(bias_type); in matchAndRewrite() local
918 bias_attr.cast<ElementsAttr>()); in matchAndRewrite()
|
| D | legalize_tfl.cc | 1494 DenseElementsAttr bias_attr; in matchAndRewrite() local
1503 bias_attr = DenseElementsAttr::get(bias_type, in matchAndRewrite()
1511 bias_attr = DenseElementsAttr::get(bias_type, in matchAndRewrite()
1515 rewriter.create<tosa::ConstOp>(op->getLoc(), bias_type, bias_attr); in matchAndRewrite()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| D | prepare_tf.cc | 365 auto bias_attr = rewriter.getZeroAttr(bias_type); in matchAndRewrite() local
367 rewriter.create<TF::ConstOp>(op->getLoc(), bias_type, bias_attr); in matchAndRewrite()
|