Searched refs:compute_fn (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | unary_ops_composition.cc | 42 ComputeFn compute_fn; member
51 void RegisterComputeFn(const string& name, ComputeFn compute_fn, int cost) { in RegisterComputeFn()
53 compute_fns[name] = {compute_fn, cost}; in RegisterComputeFn()
68 fns->push_back(reg.compute_fn); in ExportComputeFns()
117 auto compute_fn = [this, &in_flat, &out_flat, &num_fns](int64 begin, in Compute() local
136 std::move(compute_fn)); in Compute()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/rpc/ |
| D | grpc_worker_service.cc | 245 auto compute_fn = [this, call_opts, wrapped_request, in RunGraphHandler() local
254 compute_fn, done_cb); in RunGraphHandler()
256 compute_fn(done_cb); in RunGraphHandler()
277 auto compute_fn = [this, &call_opts, &call](StatusCallback done) { in RecvTensorHandlerRaw() local
285 compute_fn, done_cb); in RecvTensorHandlerRaw()
287 compute_fn(done_cb); in RecvTensorHandlerRaw()
|
| /external/tensorflow/tensorflow/contrib/layers/python/layers/ |
| D | rev_block_lib.py | 521 def _recomputing_grad_fn(compute_fn, argument
553 outputs = compute_fn(*inputs, **fn_kwargs)
613 compute_fn=fn,
|
| /external/tensorflow/tensorflow/contrib/compiler/ |
| D | jit_test.py | 49 def compute(self, use_jit, compute_fn): argument
53 r = compute_fn()
|