Home
last modified time | relevance | path

Searched refs:tensor_v (Results 1 – 1 of 1) sorted by relevance

/external/pytorch/torch/csrc/jit/passes/
Dsymbolic_shape_analysis.cpp622 SSArgument tensorShapeArg(Value* tensor_v) { in tensorShapeArg() argument
623 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()