| /external/tensorflow/tensorflow/compiler/mlir/lite/experimental/estimators/ |
| D | arithmetic_count_util.h | 73 auto bias_type = in GetArithmeticCountForConvAndFullyconnectedOp() local
75 if (bias_type && bias_type.hasStaticShape()) { in GetArithmeticCountForConvAndFullyconnectedOp()
76 *count += bias_type.getNumElements(); in GetArithmeticCountForConvAndFullyconnectedOp()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/utils/ |
| D | lstm_utils_test.cc | 53 auto bias_type = RankedTensorType::get(bias_shape, builder->getF32Type()); in createLstmCompositeFunc() local
59 SmallVector<mlir::Type, 4> input_types{input_type, weight_type, bias_type, in createLstmCompositeFunc()
|
| /external/tensorflow/tensorflow/lite/delegates/nnapi/ |
| D | nnapi_delegate_signed_quantization_test.cc | 265 tflite::TensorType bias_type = TensorType_INT32; local
268 bias_type = TensorType_INT64;
270 TensorData bias{bias_type,
|
| D | nnapi_delegate_test.cc | 634 const auto bias_type = in SetBias() local
636 SetData(bias_, bias_type, data); in SetBias()
1102 const auto bias_type = in SetBias() local
1104 SetData(bias_, bias_type, data); in SetBias()
1227 const auto bias_type = in SetBias() local
1229 SetData(bias_, bias_type, data); in SetBias()
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | conv_test.cc | 88 tflite::TensorType bias_type = TensorType_INT32; local
91 bias_type = TensorType_INT64;
93 TensorData bias{bias_type,
|
| D | depthwise_conv_test.cc | 82 tflite::TensorType bias_type = TensorType_INT32; in BaseDepthwiseConvolutionOpModel() local
84 bias_type = TensorType_INT64; in BaseDepthwiseConvolutionOpModel()
86 TensorData bias{bias_type, in BaseDepthwiseConvolutionOpModel()
|
| /external/gemmlowp/meta/generators/ |
| D | transform_kernels_common.py | 394 def __init__(self, bias_type, cc_emitter, asm_emitter): argument
396 'BiasAdd<%s>' % bias_type)
|
| /external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/ |
| D | legalize_tf.cc | 825 RankedTensorType bias_type = in matchAndRewrite() local
827 auto bias_attr = rewriter.getZeroAttr(bias_type); in matchAndRewrite()
828 auto bias = rewriter.create<tosa::ConstOp>(op->getLoc(), bias_type, in matchAndRewrite()
914 RankedTensorType bias_type = in matchAndRewrite() local
916 auto bias_attr = rewriter.getZeroAttr(bias_type); in matchAndRewrite()
917 auto bias = rewriter.create<tosa::ConstOp>(op->getLoc(), bias_type, in matchAndRewrite()
|
| D | legalize_tfl.cc | 1443 RankedTensorType bias_type = in matchAndRewrite() local
1486 if (!bias_type) { in matchAndRewrite()
1491 RankedTensorType bias_type = RankedTensorType::get( 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 | 364 auto bias_type = RankedTensorType::get({bias_dim}, elem_type); in matchAndRewrite() local
365 auto bias_attr = rewriter.getZeroAttr(bias_type); in matchAndRewrite()
367 rewriter.create<TF::ConstOp>(op->getLoc(), bias_type, bias_attr); in matchAndRewrite()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/ir/ |
| D | tfl_ops.cc | 860 const auto bias_type = in fold() local
869 (has_bias && !bias_type.getElementType().isF32())) { in fold()
875 (has_bias && !bias_type.hasStaticShape())) { in fold()
882 (has_bias && bias_type.getShape().size() != 1)) { in fold()
|
| /external/tensorflow/tensorflow/compiler/mlir/tosa/g3doc/ |
| D | legalization.md | 3697 auto bias_type = tensor<bias_shape, tosa.float>
3706 %bias_val = tosa.CONST() {value=[0] * %filter.shape[3]} : () -> bias_type
3733 auto bias_type = tensor<bias_shape, acc_dtype>
3747 %bias_val = tosa.CONST() {value=[0] * %filter.shape[3]} : () -> bias_type
3758 …TED(%input_val, %filter, %bias_val) : (input_type, tensor<%filter_val.type>, bias_type) -> acc_type
|