Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/jit/
Dxla_fusion_optimizer_test.cc186 Output var_handle = in TEST_F() local
193 root.WithOpName("assign_1"), var_handle, begin, end, strides, to_assign); in TEST_F()
195 root.WithOpName("assign_2"), var_handle, begin, end, strides, to_assign); in TEST_F()
Dresource_operation_safety_analysis_test.cc43 Output var_handle = in MakeRead() local
46 ops::ReadVariableOp(scope.WithOpName("Read" + id), var_handle, DT_FLOAT); in MakeRead()
51 Output var_handle = in MakeWrite() local
55 ops::AssignVariableOp assign_op(scope.WithOpName("Assignee" + id), var_handle, in MakeWrite()
61 Output var_handle = in MakeModify() local
65 var_handle, value_to_write); in MakeModify()
Dmark_for_compilation_pass_test.cc521 Output var_handle = in MakeRead() local
524 ops::ReadVariableOp(scope.WithOpName("Read" + id), var_handle, DT_FLOAT); in MakeRead()
526 *var_handle_op = var_handle.node(); in MakeRead()
532 Output var_handle = in MakeWrite() local
537 var_handle, value_to_write); in MakeWrite()
1268 Node* var_handle; in TEST() local
1269 Node* resource_read = MakeRead(root, "read", &var_handle); in TEST()
1273 string var_handle_name = var_handle->name(); in TEST()
1294 Node* var_handle; in TEST() local
1295 Node* resource_read = MakeRead(root, "read", &var_handle); in TEST()
[all …]
Dbuild_xla_ops_pass_test.cc109 Output var_handle = ops::VarHandleOp(scope.WithOpName("Var_" + id), DT_FLOAT, in MakeWrite() local
112 var_handle, value_to_write); in MakeWrite()
/external/tensorflow/tensorflow/c/eager/
Dc_api_test.cc1431 TFE_TensorHandle* var_handle = nullptr; in CreateVariable() local
1433 TFE_Execute(op, &var_handle, &num_retvals, status); in CreateVariable()
1442 TFE_OpAddInput(op, var_handle, status); in CreateVariable()
1463 return var_handle; in CreateVariable()
1475 TFE_TensorHandle* var_handle = CreateVariable(ctx, 12.0, status); in TEST() local
1481 TFE_OpAddInput(op, var_handle, status); in TEST()
1501 TFE_DeleteTensorHandle(var_handle); in TEST()
1516 TFE_TensorHandle* var_handle = CreateVariable(ctx, 5.0, status); in BM_ReadVariable() local
1522 TFE_OpAddInput(op, var_handle, status); in BM_ReadVariable()
1541 TFE_DeleteTensorHandle(var_handle); in BM_ReadVariable()
/external/tensorflow/tensorflow/python/eager/
Dfunction_test.py647 var_handle = compiled()
648 self.assertEqual(var_handle.dtype, dtypes.resource)
649 self.assertEqual(var_handle.shape, tensor_shape.scalar())
650 var_t = resource_variable_ops.read_variable_op(var_handle, dtype=v.dtype)
719 var_handle = compiled()
720 self.assertEqual(var_handle.dtype, dtypes.resource)
721 self.assertEqual(var_handle.shape, tensor_shape.scalar())
722 var_t = resource_variable_ops.read_variable_op(var_handle, dtype=v.dtype)
/external/tensorflow/tensorflow/python/kernel_tests/
Dresource_variable_ops_test.py944 var_handle = var._handle
948 resource_variable_ops.destroy_resource_op(var_handle,