Searched refs:create_auxiliary_tensor (Results 1 – 2 of 2) sorted by relevance
139 …ITensorInfo *logits = sketch.implementation().create_auxiliary_tensor(src->clone()->set_… in create_op()141 …ITensorInfo *sum = sketch.implementation().create_auxiliary_tensor(src->clone()->set_tensor_shape(… in create_op()
129 ITensorInfo *create_auxiliary_tensor(const ITensorInfo &tensor_info) in create_auxiliary_tensor() function