Searched refs:ResourceHandleWithLayout (Results 1 – 2 of 2) sorted by relevance
342 new ResourceHandleWithLayout(std::move(tensor), mesh, layout, *shape)); in Wrap()353 return std::unique_ptr<TensorWithLayout>(new ResourceHandleWithLayout( in Dummy()384 void ResourceHandleWithLayout::EncodeAttributes( in EncodeAttributes()395 tensorflow::Fprint128 ResourceHandleWithLayout::CacheKey() const { in CacheKey()408 void ResourceHandleWithLayout::UpdateLayout(const Layout& new_layout, in UpdateLayout()423 void ResourceHandleWithLayout::UpdateAttrs(const EmbeddingResourceAttrs& attrs, in UpdateAttrs()700 auto* resource = dynamic_cast<const ResourceHandleWithLayout*>(input); in PrepareGraphForMlir()
374 class ResourceHandleWithLayout : public TensorWithLayout {427 ResourceHandleWithLayout( in ResourceHandleWithLayout() function