Searched refs:compute_fn (Results 1 – 3 of 3) 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/python/compiler/xla/ |
| D | jit_test.py | 49 def compute(self, use_jit, compute_fn): argument
53 r = compute_fn()
|
| /external/tensorflow/tensorflow/c/experimental/saved_model/internal/ |
| D | saved_model_api_test.cc | 113 TF_ConcreteFunction* compute_fn = in TEST_P() local
124 compute_fn, compute_fn_inputs.data(), compute_fn_inputs.size(), status); in TEST_P()
|