- Sort Score
- Result 10 results
- Languages All
Results 1 - 5 of 5 for dynamic_slice (0.16 sec)
-
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) -
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) -
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) -
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) -
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)