Merge pull request #25161 from Dayananda-V:deprecated_api_patch_1
PiperOrigin-RevId: 230748461
This commit is contained in:
commit
34c412876d
@ -50,7 +50,7 @@ def tfadd_with_ckpt(out_dir):
|
|||||||
y = variables.VariableV1(constant_op.constant([0]), name='y_saved')
|
y = variables.VariableV1(constant_op.constant([0]), name='y_saved')
|
||||||
math_ops.add(x, y, name='x_y_sum')
|
math_ops.add(x, y, name='x_y_sum')
|
||||||
|
|
||||||
init_op = variables.initialize_all_variables()
|
init_op = variables.global_variables_initializer()
|
||||||
saver = saver_lib.Saver(write_version=saver_pb2.SaverDef.V1)
|
saver = saver_lib.Saver(write_version=saver_pb2.SaverDef.V1)
|
||||||
with session.Session() as sess:
|
with session.Session() as sess:
|
||||||
sess.run(init_op)
|
sess.run(init_op)
|
||||||
@ -65,7 +65,7 @@ def tfadd_with_ckpt_saver(out_dir):
|
|||||||
y = variables.VariableV1(constant_op.constant([0]), name='y_saved')
|
y = variables.VariableV1(constant_op.constant([0]), name='y_saved')
|
||||||
math_ops.add(x, y, name='x_y_sum')
|
math_ops.add(x, y, name='x_y_sum')
|
||||||
|
|
||||||
init_op = variables.initialize_all_variables()
|
init_op = variables.global_variables_initializer()
|
||||||
saver = saver_lib.Saver(name='abcprefix', write_version=saver_pb2.SaverDef.V1)
|
saver = saver_lib.Saver(name='abcprefix', write_version=saver_pb2.SaverDef.V1)
|
||||||
with session.Session() as sess:
|
with session.Session() as sess:
|
||||||
sess.run(init_op)
|
sess.run(init_op)
|
||||||
|
@ -72,7 +72,7 @@ class DenseLayerTest(test.TestCase):
|
|||||||
x = array_ops.placeholder(shape=[None, None, 3], dtype=np.float32)
|
x = array_ops.placeholder(shape=[None, None, 3], dtype=np.float32)
|
||||||
y = layers.dense(x, 3)
|
y = layers.dense(x, 3)
|
||||||
|
|
||||||
self.evaluate(variables.initialize_all_variables())
|
self.evaluate(variables.global_variables_initializer())
|
||||||
run_metadata = config_pb2.RunMetadata()
|
run_metadata = config_pb2.RunMetadata()
|
||||||
test_utils.RunWithWarmup(
|
test_utils.RunWithWarmup(
|
||||||
sess,
|
sess,
|
||||||
@ -97,7 +97,7 @@ class DenseLayerTest(test.TestCase):
|
|||||||
with jit_scope():
|
with jit_scope():
|
||||||
y = layers.dense(x, 3)
|
y = layers.dense(x, 3)
|
||||||
|
|
||||||
self.evaluate(variables.initialize_all_variables())
|
self.evaluate(variables.global_variables_initializer())
|
||||||
run_metadata = config_pb2.RunMetadata()
|
run_metadata = config_pb2.RunMetadata()
|
||||||
test_utils.RunWithWarmup(
|
test_utils.RunWithWarmup(
|
||||||
sess,
|
sess,
|
||||||
@ -126,7 +126,7 @@ class DenseLayerTest(test.TestCase):
|
|||||||
with jit_scope():
|
with jit_scope():
|
||||||
y = layers.dense(x, 3)
|
y = layers.dense(x, 3)
|
||||||
|
|
||||||
self.evaluate(variables.initialize_all_variables())
|
self.evaluate(variables.global_variables_initializer())
|
||||||
run_metadata = config_pb2.RunMetadata()
|
run_metadata = config_pb2.RunMetadata()
|
||||||
test_utils.RunWithWarmup(
|
test_utils.RunWithWarmup(
|
||||||
sess,
|
sess,
|
||||||
|
@ -227,7 +227,7 @@ class ModelOpsTest(test_util.TensorFlowTestCase):
|
|||||||
tree_ensemble_config=tree_ensemble_config.SerializeToString(),
|
tree_ensemble_config=tree_ensemble_config.SerializeToString(),
|
||||||
name="restore_tree")
|
name="restore_tree")
|
||||||
resources.initialize_resources(resources.shared_resources()).run()
|
resources.initialize_resources(resources.shared_resources()).run()
|
||||||
variables.initialize_all_variables().run()
|
variables.global_variables_initializer().run()
|
||||||
my_saver = saver.Saver()
|
my_saver = saver.Saver()
|
||||||
|
|
||||||
# Add the second tree and replace the ensemble of the handle.
|
# Add the second tree and replace the ensemble of the handle.
|
||||||
|
@ -1757,7 +1757,7 @@ class WeightedSumTest(test.TestCase):
|
|||||||
logits_core = fc_core.linear_model(features, [movies])
|
logits_core = fc_core.linear_model(features, [movies])
|
||||||
|
|
||||||
with self.cached_session() as sess:
|
with self.cached_session() as sess:
|
||||||
variables_lib.initialize_all_variables().run()
|
variables_lib.global_variables_initializer().run()
|
||||||
lookup_ops.tables_initializer().run()
|
lookup_ops.tables_initializer().run()
|
||||||
|
|
||||||
weights = column_to_variable[movies][0]
|
weights = column_to_variable[movies][0]
|
||||||
|
Loading…
Reference in New Issue
Block a user