- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 21 for var_handle (0.46 sec)
-
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/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/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) -
tensorflow/c/eager/c_api_test_util.cc
if (TF_GetCode(status) != TF_OK) return nullptr; TFE_TensorHandle* var_handle = nullptr; int num_retvals = 1; TFE_Execute(op, &var_handle, &num_retvals, status); if (TF_GetCode(status) != TF_OK) return nullptr; TFE_DeleteOp(op); if (TF_GetCode(status) != TF_OK) return nullptr; CHECK_EQ(1, num_retvals); TF_DeleteStatus(status); return var_handle; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Feb 21 22:37:46 UTC 2024 - 23.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/cc/save_variables_test.cc
%add = "tf.AddV2"(%cst, %cst) : (tensor<2xf32>, tensor<2xf32>) -> tensor<2xf32> %var_handle = "tf.VarHandleOp"() {container = "", device = "/device:CPU:0", shared_name = "var_0"} : () -> tensor<!tf_type.resource<tensor<2xf32>>> "tf.AssignVariableOp"(%var_handle, %add) : (tensor<!tf_type.resource<tensor<2xf32>>>, tensor<2xf32>) -> () return } } )mlir";
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Sep 06 19:12:29 UTC 2023 - 16.2K bytes - Viewed (0) -
tensorflow/compiler/jit/build_xla_ops_pass_test.cc
} 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), var_handle, value_to_write); return assign_op.operation.node(); }
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/tensorflow/transforms/resource_device_inference.cc
WalkResult walk_res = func_op.walk<WalkOrder::PreOrder>([&](Operation* op) { if (auto var_handle = dyn_cast<VarHandleOp>(op)) { // Record VarHandleOp's device attribute. StringRef device_attr = GetDeviceAttr(op); if (device_attr.empty()) return WalkResult::advance(); auto res = AddResourceDeviceAndEmitError(var_handle.getResource(), device_attr, op, result);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Mar 03 03:47:00 UTC 2023 - 13.3K bytes - Viewed (0) -
tensorflow/c/eager/c_api_experimental_test.cc
// created with `isolate_session_state` set to false, and update should // preserve it. { // Create a handle to `var`, using `ctx_1`. TFE_TensorHandle* var_handle = CreateVarHandle(ctx_1, remote_device, /*variable_name=*/"var"); TFE_TensorHandle* handle_1 = nullptr; int num_retvals = 1; TF_Status* status = TF_NewStatus();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Aug 03 03:14:26 UTC 2023 - 31.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/resource-alias-analysis-test.mlir
%read1 = "tf.ReadVariableOp"(%arg1) : (!tf_res) -> tensor<32xf32> // expected-remark@below {{Result #0, ID 0 : 0}} %var_handle = "tf.VarHandleOp"() {container = "c", shared_name = "v0"} : () -> !tf_res %read2 = "tf.ReadVariableOp"(%var_handle) : (!tf_res) -> tensor<32xf32> "tf.AssignVariableOp"(%arg1, %read0) : (!tf_res, tensor<32xf32>) -> ()
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Jul 21 17:19:47 UTC 2023 - 22.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/split_into_island_per_op.mlir
%read1 = "tf.ReadVariableOp"(%arg1) : (tensor<*x!tf_type.resource<tensor<32xf32>>>) -> tensor<32xf32> %var_handle = "tf.VarHandleOp"() {container = "c", shared_name = "v0"} : () -> tensor<*x!tf_type.resource<tensor<32xf32>>> %read2 = "tf.ReadVariableOp"(%var_handle) : (tensor<*x!tf_type.resource<tensor<32xf32>>>) -> 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)