Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 5 of 5 for dynamic_slice (0.16 sec)

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

      // CHECK-DAG-SAME: strides = dense<1> : tensor<1xi64>} : (tensor<2xi64>) -> tensor<1xi64>
      // CHECK: %[[RESHAPED_START2:.*]] = mhlo.reshape %[[SLICED_START2]] : (tensor<1xi64>) -> tensor<i64>
      // CHECK: %[[RESULT:.*]] = "mhlo.dynamic_slice"(%arg0, %[[RESHAPED_START1]], %[[RESHAPED_START2]]) <{slice_sizes = dense<[1, 4]> : tensor<2xi64>}> : (tensor<3x4xi32>, tensor<i64>, tensor<i64>) -> tensor<1x4xi32>
      // CHECK: return %[[RESULT]] : tensor<1x4xi32>
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Sat Apr 06 15:32:52 UTC 2024
    - 15.8K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/lite/stablehlo/transforms/tfl_stablehlo_pass.cc

          (field_name == "broadcast_dimensions" ||
           field_name == "known_expanding_dimensions" ||
           field_name == "known_nonexpanding_dimensions"))
        return true;
      if ((op_name == "stablehlo.dynamic_slice" || op_name == "stablehlo.gather") &&
          field_name == "slice_sizes")
        return true;
      if (op_name == "stablehlo.fft" && field_name == "fft_length") return true;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Jan 24 06:08:43 UTC 2024
    - 10.8K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/tfrt/tests/tf_to_corert/control_flow.mlir

      %handle_0, %flow_0 = "tf.TensorArrayV3"(%size) {clear_after_read = true, device = "/job:localhost/replica:0/task:0/device:CPU:0", dtype = f32, dynamic_size = false, element_shape = #tf_type.shape<?x100>, identical_element_shapes = true, tensor_array_name = "processed_embeddings/bidirectional_rnn/bw/bw/dynamic_rnn/input_0"} : (tensor<i32>) -> (tensor<2x!tf_type.resource<tensor<?x100xf32>>>, tensor<f32>)...
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 14 00:40:32 UTC 2024
    - 17.5K bytes
    - Viewed (0)
  4. tensorflow/compiler/jit/increase_dynamism_for_auto_jit_pass_test.cc

                              Const(zero_32))));
    
      std::vector<string> compile_time_constant_inputs;
      compile_time_constant_inputs.push_back("size");
      auto m_dynamic_slice = NodeWith(
          Op("Slice"), AssignedDevice(kDeviceName),
          Attr(kXlaCompileTimeConstantInputsAttr, compile_time_constant_inputs),
          Inputs(m_input, m_begin_s64, m_dynamic_slice_size));
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Feb 22 08:47:20 UTC 2024
    - 18.3K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/lite/tests/flatbuffer2mlir/vhlo.mlir

    //CHECK-NEXT:}
    
    func.func @dyanmic_slice(%arg0: tensor<3x3xi64>, %arg1: tensor<i64>, %arg2: tensor<i64>) -> tensor<3x3xi64> {
      %0 = "vhlo.dynamic_slice_v1"(%arg0, %arg1, %arg2) <{
        slice_sizes = #vhlo.tensor_v1<dense<[3, 3]> : tensor<2xi64>>
      }> : (tensor<3x3xi64>, tensor<i64>, tensor<i64>) -> tensor<3x3xi64>
      return %0 : tensor<3x3xi64>
    }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Mar 14 19:15:40 UTC 2024
    - 31.9K bytes
    - Viewed (0)
Back to top