Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/
Dlegalize_utils.cc479 llvm::Optional<Value> getConstTensor(PatternRewriter& rewriter, Operation* op, in getConstTensor() function
502 llvm::Optional<Value> getConstTensor<APInt>(PatternRewriter& rewriter, in getConstTensor() function
526 llvm::Optional<Value> getConstTensor<float>(PatternRewriter& rewriter, in getConstTensor() function
548 template llvm::Optional<Value> getConstTensor<int32_t>(PatternRewriter&,
Dlegalize_common.cc294 llvm::Optional<Value> a3_transpose_perm = getConstTensor<int32_t>( in convertPackOp()
341 llvm::Optional<Value> a1_transpose_perm = getConstTensor<int32_t>( in convertUnpackOp()
894 llvm::Optional<Value> a3_transpose_const = getConstTensor<int32_t>( in convertSpaceToBatchNDOp()
1115 llvm::Optional<Value> a2_transpose_perm = getConstTensor<int32_t>( in convertBatchToSpaceNDOp()
1897 llvm::Optional<Value> a3_transpose_perm = getConstTensor<int32_t>( in convertSpaceToDepthOp()
1984 llvm::Optional<Value> a3_transpose_perm = getConstTensor<int32_t>( in convertDepthToSpaceOp()
2633 getConstTensor<int32_t>(rewriter, op, perms, in convertGenericReduceOp()
3213 zp_val = getConstTensor(rewriter, op, zeropoint, shape); in convertDequantizeOp()
3224 scale_val = getConstTensor(rewriter, op, scale, shape); in convertDequantizeOp()
3326 llvm::Optional<Value> a1_filter_transpose_perm = getConstTensor<int32_t>( in convertTFConv2DCommon()
[all …]
Dlegalize_utils.h121 llvm::Optional<Value> getConstTensor(PatternRewriter& rewriter, Operation* op,
Dlegalize_tfl.cc1182 zero_bias = getConstTensor<APInt>(rewriter, op, vec, {output_channel}); in matchAndRewrite()
1185 zero_bias = getConstTensor<int32_t>(rewriter, op, vec, {output_channel}); in matchAndRewrite()
1189 zero_bias = getConstTensor<float>(rewriter, op, vec, {output_channel}); in matchAndRewrite()
1298 llvm::Optional<Value> a1_filter_transpose_perms = getConstTensor<int32_t>( in matchAndRewrite()
1406 getConstTensor<int32_t>(rewriter, op, /*vec=*/{0, 2, 1}, /*shape=*/{3}) in matchAndRewrite()
1416 getConstTensor<int32_t>(rewriter, op, /*vec=*/{0, 2, 1}, /*shape=*/{3}) in matchAndRewrite()
1996 Value min_val = getConstTensor<float>(rewriter, op, min, {}).getValue(); in matchAndRewrite()
Dlegalize_tf.cc951 llvm::Optional<Value> a1_filter_transpose_perm = getConstTensor<int32_t>( in matchAndRewrite()
1025 getConstTensor<float>(rewriter, op, vec, {output_channel}); in matchAndRewrite()