Home
last modified time | relevance | path

Searched refs:get1DConstTensor (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/
Dlegalize_utils.cc351 Value get1DConstTensor(PatternRewriter& rewriter, Operation* op, in get1DConstTensor() function
365 Value get1DConstTensor<tosa::ConstOp, float>(PatternRewriter& rewriter, in get1DConstTensor() function
378 template Value get1DConstTensor<tosa::ConstOp, int32_t>(
380 template Value get1DConstTensor<tosa::ConstOp, int64_t>(
382 template Value get1DConstTensor<TFL::ConstOp, int32_t>(
384 template Value get1DConstTensor<TFL::ConstOp, int64_t>(
Dlegalize_utils.h103 Value get1DConstTensor(PatternRewriter& rewriter, Operation* op,
Dlegalize_common.cc249 get1DConstTensor<tosa::ConstOp, int32_t>(rewriter, op, perm); in convertPackOp()
294 get1DConstTensor<tosa::ConstOp, int32_t>(rewriter, op, perm_vec); in convertUnpackOp()
956 get1DConstTensor<tosa::ConstOp, int32_t>(rewriter, op, a3_perm); in convertSpaceToBatchNDOp()
1175 get1DConstTensor<tosa::ConstOp, int32_t>(rewriter, op, a2_perm); in convertBatchToSpaceNDOp()
1734 Value a3_transpose_perm = get1DConstTensor<tosa::ConstOp, int32_t>( in convertSpaceToDepthOp()
1819 Value a3_transpose_perm = get1DConstTensor<tosa::ConstOp, int32_t>( in convertDepthToSpaceOp()
2799 get1DConstTensor<tosa::ConstOp, int32_t>(rewriter, op, {3, 0, 1, 2}); in convertTFConv2DCommon()
Dlegalize_tfl.cc1273 get1DConstTensor<tosa::ConstOp, int32_t>(rewriter, op, zero_bias_vec); in matchAndRewrite()
1278 get1DConstTensor<tosa::ConstOp, float>(rewriter, op, zero_bias_vec); in matchAndRewrite()
1382 get1DConstTensor<tosa::ConstOp, int32_t>(rewriter, op, {1, 2, 3, 0}); in matchAndRewrite()
Dlegalize_tf.cc949 get1DConstTensor<tosa::ConstOp, int32_t>(rewriter, op, {2, 0, 1, 3}); in matchAndRewrite()
1020 get1DConstTensor<tosa::ConstOp, float>(rewriter, op, zero_bias_vec); in matchAndRewrite()