Searched defs:shape_as_tensor (Results 1 – 3 of 3) sorted by relevance
16 def shape_as_tensor(x): function
181 at::Tensor InputMetadata::shape_as_tensor() const { in shape_as_tensor() function in torch::autograd::InputMetadata
79 ShapeAndValues<INDEX_TYPE> shape_as_tensor = in BuildRaggedTensorToTensorGraph() local