MCPcopy Create free account
hub / github.com/DeepRec-AI/DeepRec / computation

Function computation

tensorflow/python/ops/batch_ops.py:84–85  ·  view source on GitHub ↗
(*computation_args)

Source from the content-addressed store, hash-verified

82
83 @function.defun(autograph=autograph)
84 def computation(*computation_args):
85 return fn(*computation_args)
86
87 computation = computation.get_concrete_function(
88 *[tensor_spec.TensorSpec(dtype=x.dtype, shape=x.shape, name=str(i))

Callers 11

testBasicUnbatchV1DecoratedMethod · 0.85
outside_compilationFunction · 0.85
split_compile_and_replicateFunction · 0.85
wrapped_computationFunction · 0.85
testBasicUnbatchDecoratedMethod · 0.85
testBatchDecoratedWithCapturedInputMethod · 0.85
testBasicUnbatchDecoratedWithReshapeMethod · 0.85
_compile_internalFunction · 0.85
AddCallSiteForInstructionMethod · 0.85
FuseMethod · 0.85
CreateFusionMethod · 0.85

Calls 1

fnFunction · 0.70

Tested by 4

testBasicUnbatchV1DecoratedMethod · 0.68
testBasicUnbatchDecoratedMethod · 0.68
testBatchDecoratedWithCapturedInputMethod · 0.68
testBasicUnbatchDecoratedWithReshapeMethod · 0.68