Searched refs:GetStackShape (Results 1 – 1 of 1) sorted by relevance
40 Status GetStackShape(xla::XlaBuilder* builder, XlaResource* resource, in GetStackShape() function81 TF_RETURN_IF_ERROR(GetStackShape(builder, resource, &actual_shape)); in MaybeInitializeStack()194 OP_REQUIRES_OK(ctx, GetStackShape(b, resource, &stack_shape)); in Compile()