Home
last modified time | relevance | path

Searched refs:shape_or_status (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dstack_ops.cc42 auto shape_or_status = builder->GetShape(resource->value()); in GetStackShape() local
43 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()
Dtensor_array_ops.cc71 auto shape_or_status = builder->GetShape(resource->value()); in MaybeInitializeTensorArray() local
72 if (!shape_or_status.ok()) { in MaybeInitializeTensorArray()
73 return shape_or_status.status(); in MaybeInitializeTensorArray()
77 XLAShapeToTensorShape(shape_or_status.ValueOrDie(), &shape)); in MaybeInitializeTensorArray()
/external/tensorflow/tensorflow/compiler/tf2xla/
Dxla_op_kernel.cc518 auto shape_or_status = builder->GetShape(handle); in AssignVariableTensor() local
519 if (!shape_or_status.ok()) { in AssignVariableTensor()
520 return shape_or_status.status(); in AssignVariableTensor()
524 XLAShapeToTensorShape(shape_or_status.ValueOrDie(), &shape)); in AssignVariableTensor()