Searched defs:set_tensor_shape (Results 1 – 2 of 2) sorted by relevance
89 ITensorInfo &SubTensorInfo::set_tensor_shape(const TensorShape &shape) in set_tensor_shape() function in SubTensorInfo
351 ITensorInfo &TensorInfo::set_tensor_shape(const TensorShape &shape) in set_tensor_shape() function in TensorInfo