Searched refs:representation_shape (Results 1 – 3 of 3) sorted by relevance
390 TF_ASSIGN_OR_RETURN(xla::Shape representation_shape, in ReadVariableInputTensor()396 if (xla::ShapeUtil::Compatible(xla_shape, representation_shape)) { in ReadVariableInputTensor()529 xla::Shape representation_shape, in AssignVariableTensor()533 if (!xla::ShapeUtil::Compatible(xla_shape, representation_shape)) { in AssignVariableTensor()535 xla::AsInt64Slice(representation_shape.dimensions())); in AssignVariableTensor()537 variable->SetRepresentationShape(representation_shape); in AssignVariableTensor()
91 const absl::optional<xla::Shape>& representation_shape() const { in representation_shape() function
294 if (resource->representation_shape().has_value()) { in BuildComputation()296 elems.size(), resource->representation_shape()->layout()); in BuildComputation()