Searched refs:shape_as_tensor (Results 1 – 8 of 8) sorted by relevance
122 return grad._nested_tensor_size().is_same_size(shape_as_tensor()); in is_same_shape()152 ss << shape_as_tensor(); in incompatible_shape_error_message()181 at::Tensor InputMetadata::shape_as_tensor() const { in shape_as_tensor() function in torch::autograd::InputMetadata
101 at::Tensor shape_as_tensor() const;
16 def shape_as_tensor(x): function
79 ShapeAndValues<INDEX_TYPE> shape_as_tensor = in BuildRaggedTensorToTensorGraph() local81 AddInputFromArray<INDEX_TYPE>(shape_as_tensor.shape, in BuildRaggedTensorToTensorGraph()82 shape_as_tensor.values); in BuildRaggedTensorToTensorGraph()
297 shape_as_tensor = constant_op.constant([1], dtype=dtypes.int64)299 5, padded_shapes=shape_as_tensor)314 shape_as_tensor = array_ops.placeholder(dtypes.int64, shape=[2])316 5, padded_shapes=shape_as_tensor)
98 torch.onnx.operators.shape_as_tensor,
324 shape = torch.onnx.operators.shape_as_tensor(x)
5163 return 1 + torch.onnx.operators.shape_as_tensor(x)[0]