Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 1 of 1 for 1x64x112x112xf32 (0.09 sec)

  2. github.com/tensorflow/tensorflow

    tensorflow/compiler/mlir/lite/stablehlo/tests/optimize_layout.mlir 

    // CHECK:               : (tensor<1x114x114x64xf32>) -> tensor<1x64x114x114xf32>
// CHECK:           return %[[TPOS]] : tensor<1x64x114x114xf32>

func.func @commute_transpose_pad(
      %arg0: tensor<1x112x112x64xf32>, %padding_val: tensor<f32>)
      -> tensor<1x64x114x114xf32> {
  %tspos = stablehlo.transpose %arg0, dims = [0, 3, 1, 2]
      : (tensor<1x112x112x64xf32>) -> tensor<1x64x112x112xf32>
  %ret = stablehlo.pad %tspos, %padding_val,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 21:59:06 UTC 2024
    - 2.8K bytes
    - Viewed (0)
Back to top