Skip to main content

lookup_global_metadata

Function lookup_global_metadata 

Source
pub fn lookup_global_metadata(
    key: &ValueKey<StdTensorOp>,
) -> Result<Option<TensorMeta>, MetadataRegistryError>
Expand description

Look up a single metadata entry from the global registry.

Locks the registry briefly for a single HashMap::get + clone.

ยงExamples

use computegraph::types::ValueKey;
use tenferro_ops::ad::context::lookup_global_metadata;
use tenferro_ops::input_key::TensorInputKey;
use tenferro_ops::std_tensor_op::StdTensorOp;

let key = ValueKey::<StdTensorOp>::Input(TensorInputKey::User { id: 99 });
let meta = lookup_global_metadata(&key).unwrap();
assert!(meta.is_none());