Searched refs:var_handle (Results 1 – 7 of 7) sorted by relevance
/external/tensorflow/tensorflow/compiler/jit/ |
D | xla_fusion_optimizer_test.cc | 186 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()
|
D | resource_operation_safety_analysis_test.cc | 43 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()
|
D | mark_for_compilation_pass_test.cc | 521 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 …]
|
D | build_xla_ops_pass_test.cc | 109 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/ |
D | c_api_test.cc | 1431 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/ |
D | function_test.py | 647 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/ |
D | resource_variable_ops_test.py | 944 var_handle = var._handle 948 resource_variable_ops.destroy_resource_op(var_handle,
|