Searched refs:StagingMap (Results 1 – 1 of 1) sorted by relevance
81 class StagingMap : public ResourceBase { class338 explicit StagingMap(const DataTypeVector& dtypes, std::size_t capacity, in StagingMap() function in tensorflow::__anon07039e130111::StagingMap488 StagingMap<Ordered>** map) { in GetStagingMap()493 auto create_fn = [&ndef](StagingMap<Ordered>** ret) -> Status { in GetStagingMap()500 *ret = new StagingMap<Ordered>(dtypes, capacity, memory_limit); in GetStagingMap()505 TF_RETURN_IF_ERROR(rm->LookupOrCreate<StagingMap<Ordered>>( in GetStagingMap()516 StagingMap<Ordered>* map = nullptr; in Compute()519 typename StagingMap<Ordered>::OptionalTuple tuple; in Compute()578 StagingMap<Ordered>* map = nullptr; in Compute()581 typename StagingMap<Ordered>::Tuple tuple; in Compute()[all …]