HomeSort by relevance Sort by last modified time
    Searched refs:compute_fn (Results 1 - 4 of 4) sorted by null

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

Completed in 191 milliseconds