Home
last modified time | relevance | path

Searched refs:var_handle (Results 1 – 16 of 16) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/
Dresource-device-inference.mlir16 %var_handle = "tf.VarHandleOp"() {container = "c", shared_name = "v0", device = "/CPU:0"}
28 %id2 = "tf.Identity"(%var_handle) : (!tf_res)
56 %var_handle = "tf.VarHandleOp"() {container = "c", shared_name = "v0", device = "/TPU:1"}
59 "tf.If"(%arg1, %id0, %var_handle) {
129 %var_handle = "tf.VarHandleOp"() {container = "c", shared_name = "v0", device = "/TPU:1"}
132 "tf.While"(%arg1, %id0, %var_handle) {
211 %var_handle = "tf.VarHandleOp"() {container = "c", shared_name = "v0", device = "/TPU:1"}
213 "tf.If"(%arg1, %id0, %var_handle) {
218 "tf.If"(%arg1, %var_handle, %id0) {
274 %var_handle = "tf.VarHandleOp"() {container = "c", shared_name = "v0", device = "/CPU:0"}
[all …]
Dresource-alias-analysis-test.mlir20 %var_handle = "tf.VarHandleOp"() {container = "c", shared_name = "v0"} : () -> !tf_res
21 %read2 = "tf.ReadVariableOp"(%var_handle) : (!tf_res) -> tensor<32xf32>
Dbreakup-islands.mlir179 …%var_handle = "tf.VarHandleOp"() {container = "c", shared_name = "v0"} : () -> tensor<*x!tf.resour…
180 …%read2 = "tf.ReadVariableOp"(%var_handle) : (tensor<*x!tf.resource<tensor<32xf32>>>) -> tensor<32x…
Dside-effect-analysis-test.mlir29 …%var_handle = "tf.VarHandleOp"() {container = "c", shared_name = "v0"} : () -> tensor<*x!tf.resour…
31 …%read2 = "tf.ReadVariableOp"(%var_handle) : (tensor<*x!tf.resource<tensor<32xf32>>>) -> tensor<32x…
/external/tensorflow/tensorflow/c/eager/
Dcustom_device_test.cc122 TFE_TensorHandle* var_handle = nullptr; in TEST() local
125 TFE_Execute(op.get(), &var_handle, &num_retvals, status.get()); in TEST()
129 [var_handle]() { TFE_DeleteTensorHandle(var_handle); }); in TEST()
136 TFE_OpAddInput(op.get(), var_handle, status.get()); in TEST()
148 TFE_OpAddInput(op.get(), var_handle, status.get()); in TEST()
174 TFE_OpAddInput(op.get(), var_handle, status.get()); in TEST()
207 TFE_TensorHandle* var_handle = nullptr; in TEST() local
210 TFE_Execute(op.get(), &var_handle, &num_retvals, status.get()); in TEST()
214 [var_handle]() { TFE_DeleteTensorHandle(var_handle); }); in TEST()
221 TFE_OpAddInput(op.get(), var_handle, status.get()); in TEST()
[all …]
Dc_api_distributed_test.cc377 TFE_TensorHandle* var_handle = TestVariable(ctx, 2.0, dev2_name); in TEST() local
378 EXPECT_NE(var_handle, nullptr); in TEST()
389 TFE_OpAddInput(func, var_handle, status); in TEST()
406 TFE_DeleteTensorHandle(var_handle); in TEST()
500 TFE_TensorHandle* var_handle = TestVariable(ctx, 2.0, dev2_name); in TestDistributedFunctionCancellation() local
501 EXPECT_NE(var_handle, nullptr); in TestDistributedFunctionCancellation()
513 TFE_OpAddInput(func, var_handle, status); in TestDistributedFunctionCancellation()
535 TFE_DeleteTensorHandle(var_handle); in TestDistributedFunctionCancellation()
Dc_api_test_util.cc209 TFE_TensorHandle* var_handle = nullptr; in TestVariable() local
211 TFE_Execute(op, &var_handle, &num_retvals, status); in TestVariable()
221 TFE_OpAddInput(op, var_handle, status); in TestVariable()
244 return var_handle; in TestVariable()
Dc_api_test.cc1209 TFE_TensorHandle* var_handle = TestVariable(ctx, 12.0); in TEST() local
1215 TFE_OpAddInput(op, var_handle, status); in TEST()
1235 TFE_DeleteTensorHandle(var_handle); in TEST()
1250 TFE_TensorHandle* var_handle = TestVariable(ctx, 5.0); in BM_ReadVariable() local
1260 TFE_OpAddInput(op, var_handle, status); in BM_ReadVariable()
1274 TFE_DeleteTensorHandle(var_handle); in BM_ReadVariable()
/external/tensorflow/tensorflow/c/experimental/saved_model/core/ops/
Dvariable_ops.cc57 AbstractTensorHandle* var_handle = nullptr; in CreateUninitializedResourceVariable() local
60 absl::MakeSpan(&var_handle, num_retvals), &num_retvals)); in CreateUninitializedResourceVariable()
61 AbstractTensorHandlePtr owned_var_handle(var_handle); in CreateUninitializedResourceVariable()
/external/tensorflow/tensorflow/compiler/jit/
Dresource_operation_safety_analysis_test.cc42 Output var_handle = in MakeRead() local
45 ops::ReadVariableOp(scope.WithOpName("Read" + id), var_handle, DT_FLOAT); in MakeRead()
50 Output var_handle = in MakeWrite() local
54 ops::AssignVariableOp assign_op(scope.WithOpName("Assignee" + id), var_handle, in MakeWrite()
60 Output var_handle = in MakeModify() local
64 var_handle, value_to_write); in MakeModify()
Dmark_for_compilation_pass_test.cc654 Output var_handle = in MakeRead() local
657 ops::ReadVariableOp(scope.WithOpName("Read" + id), var_handle, DT_FLOAT); in MakeRead()
659 *var_handle_op = var_handle.node(); in MakeRead()
665 Output var_handle = in MakeWrite() local
670 var_handle, value_to_write); in MakeWrite()
1439 Node* var_handle; in TEST() local
1440 Node* resource_read = MakeRead(root, "read", &var_handle); in TEST()
1444 string var_handle_name = var_handle->name(); in TEST()
1465 Node* var_handle; in TEST() local
1466 Node* resource_read = MakeRead(root, "read", &var_handle); in TEST()
[all …]
Dbuild_xla_ops_pass_test.cc110 Output var_handle = ops::VarHandleOp(scope.WithOpName("Var_" + id), DT_FLOAT, in MakeWrite() local
113 var_handle, value_to_write); in MakeWrite()
/external/tensorflow/tensorflow/c/eager/parallel_device/
Dparallel_device_testlib.cc53 TFE_TensorHandle* var_handle = nullptr; in Create() local
55 TFE_Execute(op.get(), &var_handle, &num_retvals, status); in Create()
57 return new Variable(var_handle, type); in Create()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dresource_device_inference.cc178 if (auto var_handle = dyn_cast<VarHandleOp>(op)) { in ComputeResourceDevicesInComputation() local
182 auto res = AddResourceDeviceAndEmitError(var_handle.resource(), in ComputeResourceDevicesInComputation()
/external/tensorflow/tensorflow/python/eager/
Dfunction_test.py1228 var_handle = compiled()
1229 self.assertEqual(var_handle.dtype, dtypes.resource)
1230 self.assertEqual(var_handle.shape, tensor_shape.TensorShape([]))
1231 var_t = resource_variable_ops.read_variable_op(var_handle, dtype=v.dtype)
1305 var_handle = compiled()
1306 self.assertEqual(var_handle.dtype, dtypes.resource)
1307 self.assertEqual(var_handle.shape, tensor_shape.TensorShape([]))
1308 var_t = resource_variable_ops.read_variable_op(var_handle, dtype=v.dtype)
/external/tensorflow/tensorflow/python/kernel_tests/
Dresource_variable_ops_test.py1254 var_handle = var._handle
1258 resource_variable_ops.destroy_resource_op(var_handle,