- Sort Score
- Result 10 results
- Languages All
Results 21 - 30 of 37 for varhandle_op (0.21 sec)
-
tensorflow/compiler/mlir/tensorflow/tests/split_into_island_per_op.mlir
// CHECK: } func.func @unknown_side_effecting_op(%arg0: tensor<32xf32>) -> () { tf_executor.graph { %island = tf_executor.island { %vh0 = "tf.VarHandleOp"() {container = "c", shared_name = "v0"} : () -> tensor<*x!tf_type.resource<tensor<32xf32>>> %vh1 = "tf.VarHandleOp"() {container = "c", shared_name = "v1"} : () -> tensor<*x!tf_type.resource<tensor<32xf32>>>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Oct 30 06:52:55 UTC 2023 - 20.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tf_device_passes.td
resource variables directly.. Here is a simple example in TensorFlow where a device doubles the value of a TensorFlow resource variable and returns new value: ```mlir %resource_handle = "tf.VarHandleOp"() %1 = "tf_device.cluster"() ( { %init_value = "tf.ReadVariableOp"(%resource_handle) "tf.AssignAddVariableOp"(%resource_handle, %init_value) %new_value = "tf.ReadVariableOp"(%resource_handle)
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Apr 17 18:52:57 UTC 2024 - 12.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/breakup-islands.mlir
// CHECK: } func.func @unknown_side_effecting_op(%arg0: tensor<32xf32>) -> () { tf_executor.graph { %island = tf_executor.island { %vh0 = "tf.VarHandleOp"() {container = "c", shared_name = "v0"} : () -> tensor<*x!tf_type.resource<tensor<32xf32>>> %vh1 = "tf.VarHandleOp"() {container = "c", shared_name = "v1"} : () -> tensor<*x!tf_type.resource<tensor<32xf32>>>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Oct 31 08:59:10 UTC 2023 - 28.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/update_control_dependencies.mlir
func.func @unknown_side_effecting_op(%arg0: tensor<32xf32>) { tf_executor.graph { %outputs, %control = tf_executor.island wraps "tf.VarHandleOp"() {container = "c", shared_name = "v0"} : () -> tensor<*x!tf_type.resource<tensor<32xf32>>> %outputs_0, %control_1 = tf_executor.island wraps "tf.VarHandleOp"() {container = "c", shared_name = "v1"} : () -> tensor<*x!tf_type.resource<tensor<32xf32>>>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Nov 03 18:12:49 UTC 2023 - 25.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/host_runtime/tpu_variable_runtime_reformatting.cc
// Creates the per-device variables that represent the formatting state of each // device. llvm::SmallVector<TF::VarHandleOp, 4> CreateStateVars( const llvm::SmallDenseMap<llvm::StringRef, llvm::SmallVector<StringRef, 4>>& devices, Location loc, RankedTensorType key_type, OpBuilder* builder) { llvm::SmallVector<TF::VarHandleOp, 4> state_vars; // TODO(b/148913020): Remove this constraint once model parallelism is // supported.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 21.9K bytes - Viewed (0) -
tensorflow/compiler/jit/build_xla_ops_pass_test.cc
result); } Node* MakeWrite(const Scope& scope, Output value_to_write, const string& id) { Output var_handle = ops::VarHandleOp(scope.WithOpName("Var_" + id), DT_FLOAT, TensorShape({})); ops::AssignVariableOp assign_op(scope.WithOpName("Assignee_" + id),
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 22 08:47:20 UTC 2024 - 12.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfrt/tests/tf_to_corert/optimize_tf_control_flow_side_effect.mlir
} // CHECK-LABEL: func @no_set_stateless func.func @no_set_stateless(%arg: tensor<i32>, %cond: tensor<i1>) -> (tensor<i32>, tensor<i32>) { %handle = "tf.VarHandleOp"() { container = "", shared_name = "var" } : () -> tensor<!tf_type.resource<tensor<i32>>> "tf.AssignVariableOp"(%handle, %arg) : (tensor<!tf_type.resource<tensor<i32>>>, tensor<i32>) -> () // CHECK: tf.While
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Mar 02 01:15:55 UTC 2023 - 10.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfrt/tests/mlrt/tf_to_mlrt.mlir
// Test lowering of IfrtLoadVariableOp // CHECK-LABEL: func @ifrt_load_variable_test func.func @ifrt_load_variable_test() -> () { // CHECK: [[HANDLE:%.*]] = tf_mlrt.executeop() // CHECK-SAME: VarHandleOp %0 = "tf.VarHandleOp"() {__op_key = 1: i32, device = "/device:CPU:0", container = "", shared_name = "variable"} : () -> tensor<!tf_type.resource<tensor<1x3xf32>>> // CHECK-NEXT: "tf_mlrt.ifrt_load_variable"([[HANDLE]])
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 31 20:44:15 UTC 2024 - 24.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfrt/tests/mlrt/parallelization.mlir
// CHECK-LABEL: func private @main_stream_{{[0-9]*}} // CHECK-SAME: ([[CONTROL_FUTURE:%.*]]: !mlrt.future, [[PROMISE:%.*]]: !mlrt.promise, [[PROMISE_HANDLE:%.*]]: !mlrt.promise) // CHECK: [[HANDLE:%.*]] = "tf.VarHandleOp" // CHECK-NEXT: "tf_mlrt.tf_promise"([[PROMISE_HANDLE]], [[HANDLE]]) // CHECK: mlrt.await_control [[CONTROL_FUTURE]] // CHECK-NEXT: [[V:%.*]] = "tf.ReadVariableOp"([[HANDLE]])
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jun 08 22:07:30 UTC 2023 - 15K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/tests/merge_initializer_function_ops_to_main.mlir
%out_0, %ctl_0 = tf_executor.island wraps "tf.Const"() {value = dense<"var_0"> : tensor<1x!tf_type.string>} : () -> tensor<1x!tf_type.string> %out_1, %ctl_1 = tf_executor.island wraps "tf.VarHandleOp"() {container = "", shared_name = "var_0", device = "/device:CPU:0"} : () -> tensor<!tf_type.resource<tensor<2xf32>>>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Oct 31 08:49:35 UTC 2023 - 29.6K bytes - Viewed (0)