Searched refs:fetch_fn (Results 1 – 2 of 2) sorted by relevance
268 for tensor_type, fetch_fn, _, _ in _REGISTERED_EXPANSIONS:270 fetches, contraction_fn = fetch_fn(fetch)
1860 fetch_fn = lambda squared_tensor: ([squared_tensor.sq], lambda val: val[0]) function1864 session.register_session_run_conversion_functions(SquaredTensor, fetch_fn,1867 session.register_session_run_conversion_functions(SquaredTensor, fetch_fn,