- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for dynamic_conv (0.18 sec)
-
tensorflow/compiler/mlir/tf2xla/transforms/legalize_tf.cc
add = rewriter.create<tensor::CastOp>(loc, op.getType(), add); } rewriter.replaceOp(op, {add}); return success(); } }; // Conterts tf.Conv2D to mhlo.dynamic_conv. // TODO(disc): To recover static special case's performance with adding folding, // canonicalization func and removing ConvertConvOp. template <typename OpT, int num_spatial_dims, bool depthwise_conv = false>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 11 20:00:43 UTC 2024 - 291.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/tests/legalize-tf.mlir
} // ----- // CHECK-LABEL: conv_dynamic func.func @conv_dynamic(%arg0: tensor<?x32x32x6xf32>, %arg1: tensor<3x3x3x16xf32>) -> tensor<?x8x7x16xf32> { // CHECK: "mhlo.dynamic_conv"
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon May 06 18:46:23 UTC 2024 - 335.5K bytes - Viewed (0)