Method prune

tensorflow/tensorflow/python/eager/wrap_function.py:231–361 · view source on GitHub ↗

Extract a subgraph of this function's underlying graph. Wraps the subgraph in a new `WrappedFunction` object. Args: feeds: Input tensors to the subgraph to extract, as `Tensor` objects. fetches: Possibly-nested Python data structure containing information about outputs

(self, feeds, fetches, name=None, input_signature=None)

Source from the content-addressed store, hash-verified

source not stored for this graph (policy: none)

Callers 15

testImportedHashTableMethod · 0.80

convert_variables_to_constants_v2Function · 0.80

testPruneMethod · 0.80

testPruneRaggedMethod · 0.80

testVariableLiftingMethod · 0.80

testPruneCapturesMethod · 0.80

testGradientsOfPruneMethod · 0.80

testPruneOperationsMethod · 0.80

test_operation_returnedMethod · 0.80

_wrap_functionMethod · 0.80

function_from_graph_defFunction · 0.80

restore_variablesMethod · 0.80

Calls 10

add_captureMethod · 0.95

WrappedFunctionClass · 0.85

as_graph_elementMethod · 0.80

is_tensorMethod · 0.80

flattenMethod · 0.45

as_defaultMethod · 0.45

extendMethod · 0.45

set_shapeMethod · 0.45

valuesMethod · 0.45

_add_functionMethod · 0.45

Tested by 11

testImportedHashTableMethod · 0.64

testPruneMethod · 0.64

testPruneRaggedMethod · 0.64

testVariableLiftingMethod · 0.64

testPruneCapturesMethod · 0.64

testGradientsOfPruneMethod · 0.64

testPruneOperationsMethod · 0.64

test_operation_returnedMethod · 0.64

test_custom_pruningMethod · 0.64

test_cond_pruneMethod · 0.64

test_to_protoMethod · 0.64