Lines Matching refs:fetch

115     (sparse_tensor.SparseTensor, lambda fetch: ([
116         fetch.indices, fetch.values, fetch.dense_shape
124      lambda fetch: ([fetch.values, fetch.indices] if fetch.dense_shape is None
125                     else [fetch.values, fetch.indices, fetch.dense_shape
131     (object, lambda fetch: ([fetch], lambda fetched_vals: fetched_vals[0]),
248   def for_fetch(fetch):  argument
261     if fetch is None:
264     elif isinstance(fetch, (list, tuple)):
266       return _ListFetchMapper(fetch)
267     elif isinstance(fetch, collections_abc.Mapping):
268       return _DictFetchMapper(fetch)
269     elif _is_attrs_instance(fetch):
270       return _AttrsFetchMapper(fetch)
274         if isinstance(fetch, tensor_type):
275           fetches, contraction_fn = fetch_fn(fetch)
302     for fetch in fetches:
305             fetch, allow_tensor=True, allow_operation=True))
378     self._mappers = [_FetchMapper.for_fetch(fetch) for fetch in fetches]
417         _FetchMapper.for_fetch(fetch) for fetch in fetches.values()
444     self._mappers = [_FetchMapper.for_fetch(fetch) for fetch in values]
492     for fetch in self._fetch_mapper.unique_fetches():
493       if isinstance(fetch, ops.Operation):
494         self._assert_fetchable(graph, fetch)
495         self._targets.append(fetch)
498         self._assert_fetchable(graph, fetch.op)
499         self._fetches.append(fetch)
502       if (isinstance(fetch, ops.Tensor) and
503           (fetch.op.type == 'GetSessionHandle' or
504            fetch.op.type == 'GetSessionHandleV2')):
505         self._fetch_handles[fetch.ref()] = fetch.op.inputs[0].dtype