Searched defs:compute_fn (Results 1 – 3 of 3) sorted by relevance
42 ComputeFn compute_fn; member117 auto compute_fn = [this, &in_flat, &out_flat, &num_fns](int64_t begin, in Compute() local
45 def compute(self, use_jit, compute_fn): argument
113 TF_ConcreteFunction* compute_fn = in TEST_P() local