Searched refs:shape_or_status (Results 1 – 3 of 3) sorted by relevance
42 auto shape_or_status = builder->GetShape(resource->value()); in GetStackShape() local43 if (!shape_or_status.ok()) { in GetStackShape()44 return shape_or_status.status(); in GetStackShape()46 xla::Shape shape = shape_or_status.ValueOrDie(); in GetStackShape()
71 auto shape_or_status = builder->GetShape(resource->value()); in MaybeInitializeTensorArray() local72 if (!shape_or_status.ok()) { in MaybeInitializeTensorArray()73 return shape_or_status.status(); in MaybeInitializeTensorArray()77 XLAShapeToTensorShape(shape_or_status.ValueOrDie(), &shape)); in MaybeInitializeTensorArray()
518 auto shape_or_status = builder->GetShape(handle); in AssignVariableTensor() local519 if (!shape_or_status.ok()) { in AssignVariableTensor()520 return shape_or_status.status(); in AssignVariableTensor()524 XLAShapeToTensorShape(shape_or_status.ValueOrDie(), &shape)); in AssignVariableTensor()