Searched defs:tensor_key (Results 1 – 3 of 3) sorted by relevance
11 uint32 tensor_key = 5; field
305 TensorKey tensor_key = ntohl(wc[i].imm_data); in Run() local391 TensorKey tensor_key = next_key_++; in TransportOptionsFromTensor() local476 TensorKey tensor_key = remote_mr.tensor_key(); in TensorFromTransportOptions() local
152 def __init__(self, tensor_key, shape_keys=None, shape=None, default_value=0): argument215 tensor_key, argument