Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 1 of 1 for new_args (0.17 sec)

  1. tensorflow/compiler/mlir/tf2xla/tests/legalize-tf.mlir

    // CHECK-SAME: %[[INPUT:[a-z0-9]+]]: tensor<2x4x8x16x32x64xf32>
    func.func @strided_slice_new_axis_mask(%input: tensor<2x4x8x16x32x64xf32>) {
      // For StridedSlice input[1, tf.new_axis, ..., 8:, :10, 2:6:2, tf.new_axis]
      // New axis mask is at index 1 and 6 of sparse spec, so
      // new_axis_mask = 2^1 + 2^6 = 66
      // The ellipsis mask is applied to dim #1, #2 of input i.e, we get
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon May 06 18:46:23 UTC 2024
    - 335.5K bytes
    - Viewed (0)
Back to top