Searched defs:default_tensor (Results 1 – 3 of 3) sorted by relevance
181 def __init__(self, para_name, default_tensor, requires_grad=False, layerwise_parallel=False): argument
218 …explicit NewParameter(string para_name, tensor::TensorPtr default_tensor, bool requires_grad, bool… in NewParameter()230 tensor::TensorPtr default_tensor() { return default_tensor_; } in default_tensor() function
210 …auto default_tensor = std::make_shared<tensor::Tensor>(default_input->data_type(), default_input->… in ReflectParamBackToPython() local