- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 43 for var_handle (0.34 sec)
-
tensorflow/compiler/mlir/tensorflow/tests/decompose_resource_ops.mlir
// CHECK: [[OLD_VAR:%.*]] = "tf.ReadVariableOp"([[VAR_HANDLE]]) : (tensor<*x!tf_type.resource<tensor<*xf32>>>) -> tensor<*xf32> // CHECK: [[NEW_VAR:%.*]] = "tf.Sub"([[OLD_VAR]], [[VAR_DELTA]]) // CHECK: "tf.AssignVariableOp"([[VAR_HANDLE]], [[NEW_VAR]]) // CHECK: "tf.AssignVariableOp"([[M_HANDLE]], [[NEW_M]]) // CHECK: "tf.AssignVariableOp"([[V_HANDLE]], [[NEW_V]])
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 22 19:47:48 UTC 2024 - 51.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/tests/insert_save_op.mlir
// CHECK-DAG: %[[CST:.*]] = "tf.Const" // CHECK-DAG: %[[VAR_HANDLE:.*]] = "tf.VarHandleOp" // CHECK: "tf.AssignVariableOp"(%[[VAR_HANDLE]], %[[CST]]) // Test that a new save function that wraps the SaveV2 op is created. // CHECK: func.func private @tf_quant__save(%[[ARG:.*]]: tensor<!tf_type.string>) // CHECK: %[[VAR_HANDLE:.*]] = "tf.VarHandleOp"() // CHECK-SAME: {{.*shared_name = "var_0".*}}
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Oct 30 06:52:55 UTC 2023 - 5.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/tests/quantize-variables.mlir
%0 = "tfl.var_handle"() : () -> tensor<!tf_type.resource> %1 = "tfl.dequantize"(%arg0) : (tensor<1x2x1x3x!quant.uniform<i8:f32, 1.0>>) -> tensor<1x2x1x3xf32> "tfl.assign_variable"(%0, %1) : (tensor<!tf_type.resource>, tensor<1x2x1x3xf32>) -> () func.return %arg0 : tensor<1x2x1x3x!quant.uniform<i8:f32, 1.0>>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 09:41:17 UTC 2024 - 20.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/resource-device-inference.mlir
-> !tf_res %var_handle = "tf.VarHandleOp"() {container = "c", shared_name = "v0", device = "/TPU:1"} : () -> !tf_res "tf.If"(%arg1, %id0, %var_handle) { then_branch = @if_then_and_else, else_branch = @if_then_and_else, is_stateless = false} : (tensor<i1>, !tf_res, !tf_res) -> () "tf.If"(%arg1, %var_handle, %id0) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 17 16:01:45 UTC 2022 - 18.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/tests/insert_restore_op.mlir
// CHECK-DAG: %[[CST_0:.*]] = "tf.Const"() {{.*value = dense<1.000000e\+00> : tensor<2xf32>.*}} // CHECK-DAG: %[[VAR_HANDLE:.*]] = "tf.VarHandleOp"() {{.*shared_name = "var_0".*}} : () -> tensor<!tf_type.resource<tensor<2xf32>>> // CHECK: "tf.AssignVariableOp"(%[[VAR_HANDLE]], %[[CST_0]]) : (tensor<!tf_type.resource<tensor<2xf32>>>, tensor<2xf32>) -> ()
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Oct 30 06:52:55 UTC 2023 - 9.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/tests/merge_save_function_ops_to_main.mlir
// CHECK: tf_executor.graph // CHECK: %[[VAR_HANDLE:.*]], {{.*}} = tf_executor.island wraps "tf.VarHandleOp"() <{{{.*shared_name = "var_0".*}}}> // CHECK: %[[READ_VARIABLE:.*]], {{.*}} = tf_executor.island wraps "tf.ReadVariableOp"(%[[VAR_HANDLE]])
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Oct 30 06:52:55 UTC 2023 - 7.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/host_runtime/tfrt_ops.td
Empty strings indicate that they are non-partitioned tensors.}]>:$shape_and_slices, // VarHandles for restored tensors. Variadic<TF_Tensor>:$var_handles, // Data types for restored tensor. TypeArrayAttr: $restored_dtypes, // whether trucate is used when casting from the restored_dytpe to the type specified in VarHandle DenseBoolArrayAttr: $truncate_in_cast ); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 31 20:44:15 UTC 2024 - 6.8K bytes - Viewed (0) -
tensorflow/c/eager/custom_device_test.cc
TFE_TensorHandle* var_handle = nullptr; int num_retvals = 1; executed = false; TFE_Execute(op.get(), &var_handle, &num_retvals, status.get()); ASSERT_TRUE(TF_GetCode(status.get()) == TF_OK) << TF_Message(status.get()); ASSERT_TRUE(executed); auto handle_cleaner = tensorflow::gtl::MakeCleanup( [var_handle]() { TFE_DeleteTensorHandle(var_handle); });
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Aug 27 23:39:24 UTC 2020 - 18.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/tests/pin-ops-with-side-effects.mlir
func.func @resource_output_gets_control_node() -> tensor<!tf_type.resource<tensor<1xf32>>> { %var = "tfl.var_handle"() {container = "", shared_name = "states"} : () -> tensor<!tf_type.resource<tensor<1xf32>>> func.return %var : tensor<!tf_type.resource<tensor<1xf32>>> } // CHECK-NEXT: %[[VAR:.*]], %[[CONTROL:.*]] = tfl.control_node controls "tfl.var_handle" // CHECK-NEXT: return %[[VAR]] // CHECK-LABEL: @tfl_call_once_gets_control_node
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Aug 17 10:45:19 UTC 2022 - 5.6K bytes - Viewed (0) -
tensorflow/c/eager/c_api_distributed_test.cc
TFE_ContextSetServerDef(ctx, 0, serialized.data(), serialized.size(), status); EXPECT_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status); TFE_TensorHandle* var_handle = TestVariable(ctx, 2.0, dev2_name); EXPECT_NE(var_handle, nullptr); TFE_ContextAsyncWait(ctx, status); EXPECT_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status); const string function_def = VariableAddFunction();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 15 09:49:45 UTC 2024 - 23.5K bytes - Viewed (0)