MCPcopy Create free account
hub / github.com/tensorflow/models / function

Method function

official/modeling/tf_utils_test.py:47–49  ·  view source on GitHub ↗
()

Source from the content-addressed store, hash-verified

45
46 @tf.function
47 def function():
48 replica_value = tf.fill(shape, tf_utils.get_replica_id())
49 return tf_utils.cross_replica_concat(replica_value, axis=axis)
50
51 return function
52

Callers 15

train_stepMethod · 0.80
run_customized_training_loopFunction · 0.80
trainFunction · 0.80
distribute_datasetMethod · 0.80
step_fnMethod · 0.80
try_count_flopsFunction · 0.80
export_saved_modelFunction · 0.80
mainFunction · 0.80
mainFunction · 0.80
autoaugment.pyFile · 0.80
callMethod · 0.80
test_distort_with_boxes_run_as_tf_functionMethod · 0.80

Calls

no outgoing calls

Tested by

no test coverage detected