Searched refs:zero_bias_vec (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/ |
| D | legalize_tfl.cc | 1268 SmallVector<int64_t, 8> zero_bias_vec(output_type.getShape()[3], 0); in matchAndRewrite() local
1269 zero_bias = get1DConstTensorInt48(rewriter, op, zero_bias_vec); in matchAndRewrite()
1271 SmallVector<int32_t, 8> zero_bias_vec(output_type.getShape()[3], 0); in matchAndRewrite() local
1273 get1DConstTensor<tosa::ConstOp, int32_t>(rewriter, op, zero_bias_vec); in matchAndRewrite()
1276 SmallVector<float, 8> zero_bias_vec(output_type.getShape()[3], 0.0f); in matchAndRewrite() local
1278 get1DConstTensor<tosa::ConstOp, float>(rewriter, op, zero_bias_vec); in matchAndRewrite()
|
| D | legalize_tf.cc | 1018 SmallVector<float, 8> zero_bias_vec(output_type.getShape()[3], 0.0f); in matchAndRewrite() local
1020 get1DConstTensor<tosa::ConstOp, float>(rewriter, op, zero_bias_vec); in matchAndRewrite()
|