Searched refs:SetTypeAndShape (Results 1 – 6 of 6) sorted by relevance
100 Status SetTypeAndShape(DataType type, const TensorShape& shape);
88 Status XlaResource::SetTypeAndShape(DataType type, const TensorShape& shape) { in SetTypeAndShape() function in tensorflow::XlaResource
634 TF_RETURN_IF_ERROR(variable->SetTypeAndShape(type, shape)); in AssignVariableTensor()
76 TF_RETURN_IF_ERROR(resource->SetTypeAndShape(dtype, elem_shape)); in MaybeInitializeStack()
67 TF_RETURN_IF_ERROR(resource->SetTypeAndShape(dtype, elem_shape)); in MaybeInitializeTensorArray()
382 resource->SetTypeAndShape(update.type, update.shape)); in Compile()