Searched refs:tensor_v (Results 1 – 1 of 1) sorted by relevance
622 SSArgument tensorShapeArg(Value* tensor_v) { in tensorShapeArg() argument623 auto tt = tensor_v->type()->expect<TensorType>(); in tensorShapeArg()633 if (toIValue(tensor_v)) { in tensorShapeArg()634 auto size = constant_as<at::Tensor>(tensor_v)->sizes(); in tensorShapeArg()