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));
|