Searched refs:representation_shape (Results 1 – 2 of 2) sorted by relevance
493 TF_ASSIGN_OR_RETURN(xla::Shape representation_shape, in ReadVariableInputTensor()500 if (xla::ShapeUtil::Compatible(xla_shape, representation_shape)) { in ReadVariableInputTensor()636 TF_ASSIGN_OR_RETURN(xla::Shape representation_shape, in AssignVariableTensor()642 if (!xla::ShapeUtil::Compatible(xla_shape, representation_shape)) { in AssignVariableTensor()644 xla::AsInt64Slice(representation_shape.dimensions())); in AssignVariableTensor()646 variable->SetRepresentationShape(representation_shape); in AssignVariableTensor()
91 const absl::optional<xla::Shape>& representation_shape() const { in representation_shape() function